Я немного изменил ваш код и смог набрать 89% очков. Вы были ТАК близки! Отлично сделано с вашей стороны. Не плохо!
from sklearn.ensemble import RandomForestRegressor
from sklearn.preprocessing import OneHotEncoder
from sklearn.compose import ColumnTransformer
from sklearn.model_selection import train_test_split
import pandas as pd
dataframe = pd.read_csv("car-sales.csv")
df.head()
y = dataframe["Price"].str.replace("[\$\.\,]" , "").astype(int)
x = dataframe.drop("Price", axis=1)
cat_features = ["Make", "Colour", "Odometer", "Doors", ]
oneencoder = OneHotEncoder()
transformer = ColumnTransformer([("onehot", oneencoder, cat_features)], remainder="passthrough")
transformered_x = transformer.fit_transform(x)
transformered_x = pd.get_dummies(dataframe[cat_features])
x_train, x_test, y_train, y_test = train_test_split(transformered_x, y, test_size=.2, random_state=3)
forest = RandomForestRegressor(n_estimators=200, criterion="mse", min_samples_leaf=3, min_samples_split=3, max_depth=10)
forest.fit(x_train, y_train)
# Explained variance score: 1 is perfect prediction
print('Score: %.2f' % forest.score(x_test, y_test, sample_weight=None))
print(forest.score(x_test, y_test))
Я думаю, что это было отрицательно из-за чрезмерной подгонки из-за крайне малого количества данных.
Это прямо из документации sklearn:
и я цитирую документ:
https://scikit-learn.org/stable/modules/generated/sklearn.ensemble.RandomForestRegressor.html
https://scikit-learn.org/stable/modules/generated/sklearn.ensemble.RandomForestRegressor.html
The coefficient R^2 is defined as (1 - u/v), where u is the residual sum of
squares ((y_true - y_pred) ** 2).sum() and v is the total sum of squares
((y_true - y_true.mean()) ** 2).sum(). The best possible score is 1.0 and it
can be negative (because the model can be arbitrarily worse). A constant model
that always predicts the expected value of y, disregarding the input features,
would get a R^2 score of 0.0.
Я увеличил набор данных до 100 строк, удалил суррогатный ключ (первый столбец имеющий int id 0-99) и вот он:
введите описание изображения здесь