Skip to content

Commit

Permalink
lightgbm and catboost model and updating the model
Browse files Browse the repository at this point in the history
  • Loading branch information
praveenarjun committed Oct 4, 2024
1 parent 2f93b47 commit 6a040a7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion Stock_Price_Prediction.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -1802,7 +1802,7 @@
" f1 = f1_score(y_test > pred, y_test > pred.round())\n",
" return rmse, mae, mape, accuracy, precision, confusion, recall, f1\n",
"\n",
"# Train and evaluate LightGBM model\n",
"# Train and evaluate LightGBM model for from this directly print accuracy \n",
"model_lightgbm = lgb.LGBMRegressor()\n",
"metrics_lightgbm = train_and_evaluate_model(model_lightgbm, X_train, X_test, y_train, y_test)\n",
"print(\"LightGBM Metrics:\", metrics_lightgbm)\n",
Expand Down

0 comments on commit 6a040a7

Please sign in to comment.