training data without x features

models/catboost_model.py

@@ -11,7 +11,15 @@ from tqdm import tqdm
 
 
 class CAT_BOOST:
-    def __init__(self, data_frame, params={}, n_split_kfold=5, test_size=0.15, seed=42):
+    def __init__(
+        self,
+        data_frame,
+        params={},
+        n_split_kfold=5,
+        test_size=0.15,
+        seed=42,
+        output_file_tuning="cat_boost_tuning_results.csv",
+    ):
         self.data_frame = data_frame
         self.params = params
         self.n_split_kfold = n_split_kfold
@@ -35,6 +43,8 @@ class CAT_BOOST:
         self.kmeans_estimator = self.params.get("kmeans_estimator", 5)
         self.tuning_results = None
 
+        self.output_file_tuning = output_file_tuning
+
     def preprocess(self):
         self.scaling_method = self.params.get("scaling_method", None)
 
@@ -208,6 +218,6 @@ class CAT_BOOST:
         df_tuning = pandas.concat(
             [df_tuning.drop(columns=["metrics"]), metrics_df], axis=1
         )
-        df_tuning.to_csv("cat_boost_tuning_results.csv", index=False)
+        df_tuning.to_csv(self.output_file_tuning, index=False)
 
         return