training data without x features

models/compare_models.py

@@ -2,9 +2,9 @@ import pandas
 from catboost_model import CAT_BOOST
 from lightgbm_model import LIGHT_GBM
 
-data_frame = pandas.read_csv("./data/Ketamine_icp_no_missing.csv")
+data_frame = pandas.read_csv("../data/Ketamine_icp_no_missing.csv")
 
-cat_boost_results = pandas.read_csv("./cat_boost_tuning_results.csv")
+cat_boost_results = pandas.read_csv("./cat_boost_tuning_results_no_x.csv")
 lgbm_results = pandas.read_csv("./lightgbm_tuning_results.csv")
 
 
@@ -44,12 +44,12 @@ lgbm_test_metrics_clean = clean_metrics(lgbm_test_metrics)
 
 comparison_df = pd.DataFrame(
     [
-        {"model": "catboost", **cat_test_metrics_clean},
+        {"model": "lightgbm_no_x", **cat_test_metrics_clean},
         {"model": "lightgbm", **lgbm_test_metrics_clean},
     ]
 )
 
-comparison_filename = "comparison_catboost_lightgbm.csv"
+comparison_filename = "comparison_lightgbm_no_x_vs_lightgbm.csv"
 comparison_df.to_csv(comparison_filename, index=False)
 
 print(f"Comparison saved to: {comparison_filename}")