user_train_data

main_userupload.py

@@ -43,7 +43,11 @@ class LightGBM(object):
             "address",
             "tag1",
             "tag2",
-            "tag3"
+            "tag3",
+            "rov_30",
+            "rov_3",
+            "ros_30",
+            "ros_3"
         ]
         self.str_columns = ["channel", "address", "tag1", "tag2", "tag3"]
         self.float_columns = [
@@ -57,7 +61,11 @@ class LightGBM(object):
             "user_return_videos_3",
             "user_return_3",
             "user_share_3",
-            "user_view_3"
+            "user_view_3",
+            "rov_30",
+            "rov_3",
+            "ros_30",
+            "ros_3"
         ]
         self.split_c = 0.7
         self.yc = 0.8
@@ -130,7 +138,7 @@ class LightGBM(object):
         Load dataset
         :return:
         """
-        path = "data/train_data/spider_train_20240402"
+        path = "data/train_data/spider_train_20240402.json"
         x, y = self.read_data(path)
         train_size = int(len(x) * self.split_c)
         X_train, X_test = x[:train_size], x[train_size:]