如何预测蟒蛇的结果？

Question

我有以下代码，其中我从4个输入值预测一个值：

import numpy as np

from sklearn.naive_bayes import GaussianNB

from sklearn.model_selection import train_test_split

from sklearn.neural_network import MLPClassifier

data = np.loadtxt('C:/Users/hedeg/Desktop/RulaSoftEdgePrediction.txt')

X_train = np.array(data[0:3500,0:4])

y_train = np.array(data[0:3500,4])

X_test = np.array(data[3500::,0:4])

y_test = np.array(data[3500::,4])

clf = MLPClassifier(solver='lbfgs', alpha=1e-5,hidden_layer_sizes=(5, 2), random_state=1)

clf.fit(X_train, y_train)

我收到此错误消息：

raise ValueError("Unknown label type: %s" % repr(ys))

ValueError: Unknown label type: (array([1. , 1.1, 1.2, ..., 3. , 3. , 3. ]),)

我该如何解决这个问题？

Answer 1

尝试使用这个：from sklearn.linear_model import LogisticRegressionfrom sklearn.datasets import make_blobs# generate 2d classification datasetX, y = make_blobs(n_samples=100, centers=2, n_features=2, random_state=1)# fit final modelmodel = LogisticRegression()model.fit(X, y)# example of training a final classification modelfrom sklearn.linear_model import LogisticRegressionfrom sklearn.datasets import make_blobs# generate 2d classification datasetX, y = make_blobs(n_samples=100, centers=2, n_features=2, random_state=1)# fit final modelmodel = LogisticRegression()model.fit(X, y)