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之外)。
答案 0 :(得分:2)
如果您想使用多种功能训练模型,请参见以下示例:
FEATURES_ARR = ['feature1', 'feature2', 'feature3']
lrModel = LinearRegression(normalize=True)
lrModel.fit(X_train[FEATURES_ARR], Y_train)