Изменения

Перейти к: навигация, поиск

Вариации регрессии

912 байт добавлено, 23:38, 23 января 2019
Пример кода для Scikit-learn
===Пример кода для Scikit-learn===
<font color = green># Импорт библиотек</font>
'''from''' sklearn.linear_model '''import''' Ridge
'''from''' sklearn.model_selection '''import''' train_test_split
X = ... <font color = green># загрузка X</font>
y = ... <font color = green># загрузка y</font>
<font color = green># разделение данных на train и test</font>
train_X, test_X, train_y, test_y = train_test_split(X, y)
ridge_regression = Ridge(alpha=0.1) <font color = green># alpha {{---}} величина регуляризации</font>
<font color = green># обучение</font>
ridge_regression.fit(train_X, train_y)
<font color = green># предсказание результата</font>
''print''(ridge_regression.predict(test_X))
<font color = green># вывод точности предсказания</font>
''print''(ridge_regression.score(test_X, test_y))
==Лассо-регрессия==
276
правок

Навигация