def train(self): self.set_conf() local_fs = LocalFileSystem.DEFAULT_FS tmp_path = tempfile.gettempdir() save_path = local_fs + tmp_path + "/model" log_path = local_fs + tmp_path + "/GBDTlog" input_path = "data/exampledata/GBDTLocalExampleData/agaricus.txt.train" output_path = "data/output" self.conf[AngelConf.ANGEL_TRAIN_DATA_PATH] = input_path self.conf[AngelConf.ANGEL_SAVE_MODEL_PATH] = output_path self.conf[AngelConf.ANGEL_SAVE_MODEL_PATH] = save_path # Set log path self.conf[AngelConf.ANGEL_LOG_PATH] = log_path # Set actionType train self.conf[AngelConf.ANGEL_ACTION_TYPE] = MLConf.ANGEL_ML_TRAIN runner = GBDTRunner() runner.train(self.conf)
def train(self): self.set_conf() runner = GBDTRunner() runner.train(self.conf)