我想使用scikit-neuralnetwork进行NN分类，我有5个类，所以在输出层，我有单位= 5但是我得到这个错误：数据集大小和输出层中的单位不匹配，我重新塑造了我的y_train和根据文档将"Sigmoid"函数应用于输出图层：http://scikit-neuralnetwork.readthedocs.io/en/latest/guide_model.html#classification
如果要进行多标签分类，只需使用具有多个维度的y数组进行拟合，例如：形状（N，3）用于三个不同的类别 . 然后，确保最后一层是Sigmoid .
y_train形状是：（2115,5）X_train形状是：（2115,343）这是代码：
import sknn.mlp as mlp
from sknn.mlp import Classifier
ip_layer = mlp.Layer('Sigmoid', units=1)
hidden_layer = mlp.Layer('Tanh', units=100)
op_layer = mlp.Layer('Sigmoid', units=5)
nn = Classifier(
[ip_layer, hidden_layer, op_layer],
n_iter=10000
)
nn.fit(X_train, y_train)
1 回答
输入层有一个单元 .
如果你设置
ip_layer = mlp.Layer('Sigmoid', units=343)它应该工作 .