using luigi.build

vaaaaanquish · Mar 2, 2020 · af9760f · af9760f
1 parent 68b1d3f
commit af9760f
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/nishika/model/train_bi_catb.py b/nishika/model/train_bi_catb.py
@@ -1,6 +1,6 @@
 from logging import getLogger
 
-import gokart
+import luigi
 import redshells
 from luigi.util import inherits
 from nishika.model.lgbm_feature_selection import MakeTrainSelectionFeatureData
@@ -32,7 +32,7 @@ def requires(self):
         data = self.clone(MakeTrainSelectionFeatureData)
         task = self.clone(TrainOptimizeCATB)
         if not task.complete():
-            gokart.build([task], local_scheduler=True)
+            luigi.build([task], local_scheduler=True)
         param = task.output().load()
         return redshells.train.TrainClassificationModel(rerun=True,
                                                         train_data_task=data,

diff --git a/nishika/model/train_bi_lgbm.py b/nishika/model/train_bi_lgbm.py
@@ -1,6 +1,6 @@
 from logging import getLogger
 
-import gokart
+import luigi
 import redshells
 from luigi.util import inherits
 from nishika.model.lgbm_feature_selection import MakeTrainSelectionFeatureData
@@ -32,7 +32,7 @@ def requires(self):
         data = self.clone(MakeTrainSelectionFeatureData)
         task = self.clone(TrainSelectedOptimizeLGBM)
         if not task.complete():
-            gokart.run([task], local_scheduler=True)
+            luigi.build([task], local_scheduler=True)
         param = task.output().load()
         return redshells.train.TrainClassificationModel(rerun=True,
                                                         train_data_task=data,

diff --git a/nishika/model/train_bi_xgb.py b/nishika/model/train_bi_xgb.py
@@ -1,7 +1,7 @@
 from logging import getLogger
 
 import redshells
-import gokart
+import luigi
 from luigi.util import inherits
 from nishika.model.lgbm_feature_selection import MakeTrainSelectionFeatureData, MakeTrainFeatureData
 from nishika.utils.base import Nishika
@@ -34,7 +34,7 @@ def requires(self):
         data = self.clone(MakeTrainSelectionFeatureData)
         task = self.clone(TrainOptimizeXGB)
         if not task.complete():
-            gokart.build([task], local_scheduler=True)
+            luigi.build([task], local_scheduler=True)
         param = task.output().load()
         return redshells.train.TrainClassificationModel(rerun=True,
                                                         train_data_task=data,
@@ -71,7 +71,7 @@ def requires(self):
         data = self.clone(MakeTrainFeatureData)
         task = self.clone(TrainNonSelectionOptimizeXGB)
         if not task.complete():
-            gokart.build([task], local_scheduler=True)
+            luigi.build([task], local_scheduler=True)
         param = task.output().load()
         return redshells.train.TrainClassificationModel(rerun=True,
                                                         train_data_task=data,