【发布时间】:2020-01-28 17:33:45
【问题描述】:
我用 Keras 制作了我的第一个 ANN。这是一个具有 5 个特征和 1 个输出的线性回归模型。我用“MSE”和“损失函数”做了一个图,这些就是结果。我们能说它是一个很好的模型吗?此外 R^2 = 0.91 。 这是正确的方法吗?
classifier = Sequential()
classifier.add(Dense(5, input_dim=5,kernel_initializer='normal',activation='relu'))
classifier.add(Dense(5, activation='relu'))
classifier.add(Dense(1,activation='linear'))
classifier.compile(loss='mse', optimizer='adam', metrics=['mse','mae'])
history = classifier.fit(X_train, y_train, batch_size=10, validation_data=(X_test, y_test), epochs=200, verbose=0)
y_pred=classifier.predict(X_test)
train_mse=classifier.evaluate(X_train, y_train, verbose=0)
plt.title('Loss / Mean Squared Error')
plt.plot(history.history['loss'], label='train')
plt.plot(history.history['val_loss'], label='test')
plt.legend()
plt.show()
【问题讨论】:
标签: python machine-learning keras neural-network regression