@@ -30,7 +30,7 @@ class LightGBM(object):
"tag3",
"tag4"
]
- self.str_columns = ["tag1", "tag2", "tag3", "tag4"]
+ self.str_columns = ["tag1", "tag2"]
self.split_c = 0.75
self.yc = 0.8
self.model = "models/lightgbm_0408_all_tags.bin"