Это мой код:
Код: Выделить всё
y_target = data['price']
X_data = data.drop(['price'], axis=1)
X_dummies = pd.get_dummies(X_data, dtype=int)
Код: Выделить всё
from sklearn.linear_model import LinearRegression
from sklearn.metrics import mean_squared_error, r2_score
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import MinMaxScaler
from time import time
X_train, X_test, y_train, y_test = train_test_split(
X_dummies, y_target, test_size=0.3, random_state=int(time()))
X_scaler = MinMaxScaler().fit(X_train)
X_train_scaled = X_scaler.transform(X_train)
X_test_scaled = X_scaler.transform(X_test)
lr_model = LinearRegression()
lr_model.fit(X_train_scaled, y_train)
y_pred = lr_model.predict(X_test_scaled)
print("RMSE:", np.sqrt(mean_squared_error(y_test, y_pred)))
print("R2 Score:", lr_model.score(X_test_scaled, y_test))
Набор данных: https://www.kaggle.com/datasets/toramky ... le-dataset
Подробнее здесь: https://stackoverflow.com/questions/781 ... regression