From 6edd0bf298ab7b1f6e315af28717b5d4f940da33 Mon Sep 17 00:00:00 2001 From: you-n-g Date: Sun, 3 Apr 2022 20:43:02 +0800 Subject: [PATCH] fix ddgda run all bug & pylint (#1031) --- examples/benchmarks_dynamic/DDG-DA/workflow.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/examples/benchmarks_dynamic/DDG-DA/workflow.py b/examples/benchmarks_dynamic/DDG-DA/workflow.py index f7acac075..2d7427cfd 100644 --- a/examples/benchmarks_dynamic/DDG-DA/workflow.py +++ b/examples/benchmarks_dynamic/DDG-DA/workflow.py @@ -9,13 +9,10 @@ from qlib.data.dataset.handler import DataHandlerLP import pandas as pd import fire import sys -from tqdm.auto import tqdm -import yaml import pickle from qlib import auto_init -from qlib.model.trainer import TrainerR, task_train +from qlib.model.trainer import TrainerR from qlib.utils import init_instance_by_config -from qlib.workflow.task.gen import RollingGen, task_generator from qlib.workflow import R from qlib.tests.data import GetData @@ -47,9 +44,10 @@ class DDGDA: rb = RollingBenchmark(model_type="gbdt") task = rb.basic_task() - model = init_instance_by_config(task["model"]) - dataset = init_instance_by_config(task["dataset"]) - model.fit(dataset) + with R.start(experiment_name="feature_importance"): + model = init_instance_by_config(task["model"]) + dataset = init_instance_by_config(task["dataset"]) + model.fit(dataset) fi = model.get_feature_importance()