如何在sklearn库中为训练模型提供多个输入?

时间:2015-03-09 06:35:24

标签: python scikit-learn

X_train,X_test,Y_train,Y_test = train_test_split(T,Yout,test_size = 0.20)
clf.fit(X_train,Y_train)

我需要使用3个参数训练模型,即clf.fit(X_train,X_train[3], Y_train)。但它显示错误拟合(X.shape [0],y.shape [0]))。想要使训练依赖于X_train [3]。 建议在python中是否有任何其他命令用于训练(即除了fit和train_test_split之外)。

1 个答案:

答案 0 :(得分:2)

如果您想使用多种功能训练模型,请参见以下示例:

FEATURES_ARR = ['feature1', 'feature2', 'feature3']
lrModel = LinearRegression(normalize=True)
lrModel.fit(X_train[FEATURES_ARR], Y_train)