首页 > 解决方案 > 未知标签类型:“连续”sklearn LogisticRegression

问题描述

我有以下代码来测试 sklearn python 库的 ML 算法:

我在 LogisticRegression 调用中收到以下错误:

from sklearn.linear_model import LogisticRegression
from sklearn.svm import SVC
from sklearn.ensemble import RandomForestClassifier
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from sklearn import preprocessing
from sklearn import utils
y=np.array([ 2.664, 3.108, 7.104,  0.444, 1.776, 1.776, 1.776, 4.440])

x=np.array([[ 2.26414],
           [ 3.4893 ],
           [ 7.08023],
           [11.3826 ],
           [ 5.01367],
           [ 3.13454],
           [ 2.7636 ],
           [ 1.36057]])
from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(x,y,test_size=0.2)

lr = LogisticRegression()
lr.fit(X_train, y_train)

lr.score(X_test, y_test)
ValueError  
----> 2 lr.fit(X_train, y_train)
ValueError: Unknown label type: 'continuous'

标签: pythonlogistic-regressionsklearn-pandas

解决方案


推荐阅读