Python Bytes - Linear Regression Prediction #python #datascience #coding Code in Description

#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'] 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) new_RM = np.array([2.5]).reshape(-1,1) print("Prediction: " + str(model.predict(new_RM)))