Hej! Sitter och testar lite med MLPRegressor från sklearn. Jag märkte att om jag kör cross validation så får jag en modell med bättre score än om jag kör utan. Min fråga är hur jag hämtar ut just den bästa modellen och använder mig av den.
Jag använder Tesla datasetet som finns på kaggle.com.
Som output får jag:
Jag vill alltså använda mig av modell nummer 2 (med index 1).
Enligt mitt kursmaterial står det så här:
Och deras förslag på att göra detta är:
Men längre än så kommer jag inte, och finns ingen mer information i kursen. Kör jag sista raden med kod får jag: KeyError: 'estimators', och vet inte hur jag ska fortsätta härifrån.
Någon som har några tips? Eller har jag missförstått något?
Kod:
import pandas as pd
from sklearn.neural_network import MLPRegressor
from sklearn.model_selection import cross_validate
dataset = pd.read_csv("TSLA.csv")
y = dataset["Close"]
X = dataset.drop(["Close", "Date"], axis=1)
model = MLPRegressor(random_state=1,
max_iter=5000,
verbose=True,
hidden_layer_sizes=(10, ))
cv_results = cross_validate(model, X, y,
cv=10, verbose=3,
scoring="neg_mean_squared_error",
return_estimator=True)
for current in cv_results.keys():
print("{0}:".format(current))
print(cv_results[current])
print()
Jag använder Tesla datasetet som finns på kaggle.com.
Som output får jag:
Citat:
test_score:
[ -2.27087831 -0.62080632 -7.70970914 -44.14671272 -14.9313784
-25.50519682 -13.61407291 -35.84139504 -50.3002465 -57.67208527]
[ -2.27087831 -0.62080632 -7.70970914 -44.14671272 -14.9313784
-25.50519682 -13.61407291 -35.84139504 -50.3002465 -57.67208527]
Jag vill alltså använda mig av modell nummer 2 (med index 1).
Enligt mitt kursmaterial står det så här:
Citat:
With both scores and models available, a data engineer may review the scores and select a model with the desired score. The model may then be retrieved using the model's index (or score).
Och deras förslag på att göra detta är:
Kod:
# Specificy the index of the model
# to retrieve from the list of estimators.
model = cv_results["estimators"][1]
Men längre än så kommer jag inte, och finns ingen mer information i kursen. Kör jag sista raden med kod får jag: KeyError: 'estimators', och vet inte hur jag ska fortsätta härifrån.
Någon som har några tips? Eller har jag missförstått något?