瀏覽代碼

仅通过标题tag 分析全部数据

罗俊辉 1 年之前
父節點
當前提交
935a840449
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      main.py

+ 1 - 1
main.py

@@ -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"