數據集下載
csdn資源:https://download.csdn.net/download/qq_35008279/11122908
百度雲鏈接:https://pan.baidu.com/s/1duoc2qaEyowuwWUs7gRs8Q 密碼:pe9y
from keras.models import Sequential
from keras.layers import Dense
import numpy as np
# 設定隨機數種子
np.random.seed(7)
# 導入數據
dataset = np.loadtxt('amp_full_num21.csv', delimiter=',')
# 分割輸入x和輸出Y
x = dataset[:, 1 : 200]
Y = dataset[:, 0]
# 創建模型
model = Sequential()
model.add(Dense(12, input_dim=199, activation='relu'))
model.add(Dense(8, activation='relu'))
model.add(Dense(1, activation='sigmoid'))
# 編譯模型
model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
# 訓練模型
model.fit(x=x, y=Y, epochs=150, batch_size=10)
# 評估模型
scores = model.evaluate(x=x, y=Y)
print('\n%s : %.2f%%' % (model.metrics_names[1], scores[1]*100))
結果