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