首页 文章

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

提问于
浏览
3

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 回答

  • 1

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

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

相关问题