# Sk-Learn содержит модель линейной регрессии
from
sklearn .linear_model
import
LinearRegression
# Инициализировать модель линейной регрессии
предиктор
=
LinearRegression (n_jobs
=
-
1
)
# Заполняем модель данными
Predictor.fit (X
=
TRAIN_INPUT, y
=
TRAIN_OUTPUT)
Тестирование данных
Тестирование выполняется вручную. Тестирование может быть выполнено с использованием некоторых случайных данных и тестирования, если модель дает правильный результат для ввода.
# Случайные тестовые данные
X_TEST = [[10, 20, 30]]
# Предсказать результат X_TEST, который содержит данные тестирования
result = Predictor.predict (X = X_TEST)
# Предсказать коэффициенты
коэффициенты = Predictor.coef_
# Печать результат, полученный для тестовых данных
print (
`Результат: {} Коэффициенты: {}`
.format (результат, коэффициенты))
Результат приведенных выше тестовых данных должен быть 10 + 20 * 2 + 30 * 3 = 140.
Выход
Результат: [140.] Коэффициенты: [ 1. 2. 3.]