具有 MLPRegression 错误的神经网络“y_true 和 y_pred 有不同数量的输出”

问题描述 投票:0回答:0

在python中导入库后,我做了一个简单的神经网络,

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.25, random_state=42)
model_nnr = MLPRegressor(
    hidden_layer_sizes = (30,30,30,30,30),
    activation = 'logistic',
    random_state = 0
).fit(X_train, y_train)

pred_nnr = model_nnr.predict(X_test)

print('r2_score:',r2_score(X_test, pred_nnr))
print('Mean Squared Error:',mean_squared_error(X_test, pred_nnr))

但是我遇到了以下错误,

ValueError: y_true and y_pred have different number of output (9!=1)

你能帮我解决这个问题吗?我有 9 个输入列和 1 个输出列。我搜索了错误并发现了这个帖子

您需要将输出层中的神经元数量更改为 1,或者添加一个只有 1 个神经元的新输出层。

但仍然找不到如何解决我的问题。

python neural-network
© www.soinside.com 2019 - 2024. All rights reserved.