Как создать список моделей, чтобы присоединить к fit?
Мне нужно вызывать несколько моделей, и решила попробовать их положить в список, но скормить их потом не получается для обучения модели. Может есть другой способ?
from sklearn.preprocessing import LabelEncoder
func = ['LinearRegression()', 'DecisionTreeClassifier(random_state=241)','KNeighborsClassifier(n_neighbors=5)', 'LogisticRegression()', 'GaussianNB()', 'SVC()']
def fit_predict_scaler(X_train, y_train, X_test, y_test, lr):
lr = LabelEncoder().fit(lr).transform(lr)
lr.fit(X_train, y_train)
prediction = lr.predict(X_test)
error = mean_absolute_error(prediction, y_test)
return error
fit_predict_scaler(X_train, y_train, X_test, y_test, func[0])
ValueError: bad input shape ()
Ответы (1 шт):
Автор решения: Victor VosMottor thanks Monica
→ Ссылка
func = [LinearRegression(), DecisionTreeClassifier(random_state=241),KNeighborsClassifier(n_neighbors=5), LogisticRegression(), GaussianNB(), SVC()]