Question

我想使用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)

Answer 1

输入图层有一个单元。

如果设置ip_layer = mlp.Layer('Sigmoid', units=343)，它应该有用。

AssertionError：数据集大小与输出层中的单位不匹配

1 个答案: