Скачать или смотреть Python Bytes - Sklearn Linear Regression

#Coded by Andrew C

import pandas as pd
import numpy as np
from sklearn import datasets
from sklearn.linear_model import LinearRegression
from sklearn.model_selection import train_test_split

wine_dataset = datasets.load_wine()

wine = pd.DataFrame(wine_dataset.data, columns=wine_dataset.feature_names)

X = wine[['ash']]
Y = wine['alcohol']

#Fit Data
X_train, X_test, Y_train, Y_test = train_test_split(X, Y, test_size = 0.3, random_state=1)
model = LinearRegression()
model.fit(X_train, Y_train)

#Print Results
print("Intercept: " + str(model.intercept_.round(2)))
print("Slope: " + (str(model.coef_.round(2))))
print("Formula: Y = " + str(model.intercept_.round(2)) + " + " + str(model.coef_.round(2)) + " * X")