【发布时间】:2016-12-03 13:03:22
【问题描述】:
我在玩 Keras 和一个虚拟数据集。我想看看与带有 RBF 内核的标准 SVM 相比,神经网络能做得更好。任务很简单:预测集合 {0,1,2} 中 20 维向量的类别。
我注意到神经网络做得很糟糕。 SVM 的正确率约为 90%,而神经网络的正确率为 40%。我在我的代码中做错了什么?这很可能是我的错误,但在 NN 上尝试了各种参数几个小时后,我放弃了。
代码
from sklearn.datasets import make_multilabel_classification
from sklearn.svm import SVC
from keras.models import Sequential
from keras.layers import Dense, Dropout, Activation
from keras.optimizers import SGD, RMSprop
from keras.utils import np_utils
from keras.datasets import mnist
# generate some data
dummyX, dummyY = make_multilabel_classification(n_samples=4000, n_features=20, n_classes=3)
# neural network
model = Sequential()
model.add(Dense(20, input_dim=20))
model.add(Activation('relu'))
model.add(Dropout(0.5))
model.add(Dense(20))
model.add(Activation('relu'))
model.add(Dropout(0.5))
model.add(Dense(3))
model.add(Activation('softmax'))
model.compile(loss='mean_squared_error',
optimizer='sgd',
metrics=['accuracy'])
X_train, X_test, y_train, y_test = train_test_split(dummyX, dummyY, test_size=0.20, random_state=42)
model.fit(X_train, y_train,nb_epoch=20, batch_size=30, validation_data=(X_test, y_test))
# Epoch 20/20
# 3200/3200 [==============================] - 0s - loss: 0.2469 - acc: 0.4366 - val_loss: 0.2468 - val_acc: 0.4063
# Out[460]:
# SVM - note that y_train and test are binary label. I haven't included the multi class converter code here for brevity
svm = SVC()
svm.fit(X_train, y_train)
svm.score(X_test, y_test)
# 0.891249
TL;DR
制作虚拟数据;神经网络很烂; SVM 踢了它。请帮忙
【问题讨论】:
-
神经网络对规模敏感。尝试标准化您的数据。
-
为什么 MSE 会丢失?交叉熵更适合分类。
标签: python neural-network gpu theano keras