使用MLPClassifier构建神经网络以预测多个输出

时间:2019-06-10 02:04:53

标签: python neural-network

我现在正在实现MLPClassifier来尝试神经网络,但是我不确定是否可以将多个类作为输出。 这是我的输出,如下所示:

y = array([1, 1, 2, 2, 1, 1, 2, 2, 2, 3, 3, 1, 1, 1], dtype=int64)

我想做的是创建多个类(1,2,3),并希望检查是否可以使用神经网络进行预测。

我不确定自己走的路是否正确。

我只能放置单个二进制输出,例如[0,1],还是可以放置神经网络可以预测的多个输出?

X_train, X_test, y_train, y_test = train_test_split(X, y,test_size=0.70, random_state=0)
mlp = MLPClassifier(solver='lbfgs',random_state=0, hidden_layer_sizes[1],alpha=0.001,max_iter=100)
mlp.fit(X_train, y_train)
y_pred=mlp.predict(X_test)

1 个答案:

答案 0 :(得分:0)

为什么不尝试打印y_pred以显示分类器是执行多类分类还是二进制分类?我认为它将执行多类分类。