diverse wijzigingen voor Linux implementatie
This commit is contained in:
@@ -4,7 +4,7 @@ import holidays
|
||||
import xgboost as xgb
|
||||
from sklearn.metrics import mean_absolute_error
|
||||
|
||||
print("Start training Model 1.5...")
|
||||
print("Start training Model 1.5 for Linux...")
|
||||
|
||||
# --- 1. DATA LADEN ---
|
||||
# We gebruiken de CSV die we al hadden
|
||||
@@ -83,7 +83,7 @@ xgb_model_v1_5 = xgb.XGBRegressor(
|
||||
early_stopping_rounds=50
|
||||
)
|
||||
|
||||
print("\nModel v1.5 aan het trainen...")
|
||||
print("\nModel v1.5 for Linux aan het trainen...")
|
||||
xgb_model_v1_5.fit(
|
||||
X_train, y_train,
|
||||
eval_set=[(X_test, y_test)],
|
||||
@@ -97,6 +97,6 @@ mae_v1_5 = mean_absolute_error(y_test, voorspellingen_v1_5)
|
||||
print(f"\nNieuwe Model (v1.5) MAE: {mae_v1_5:.4f}")
|
||||
|
||||
# --- 7. MODEL OPSLAAN ---
|
||||
MODEL_FILE = 'price_forecast_model_v1_5.json'
|
||||
MODEL_FILE = 'price_forecast_model_1.5.linux.json'
|
||||
xgb_model_v1_5.save_model(MODEL_FILE)
|
||||
print(f"\nModel 1.5 opgeslagen als: {MODEL_FILE}")
|
||||
Reference in New Issue
Block a user