Skip to content

GitLab

Unverified Commit f98ee672 authored by liuzhe-lz's avatar liuzhe-lz Committed by GitHub
Browse files

[v2.0] Refactor code hierarchy (part 1) (#2962)

parent f1105409
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
src/sdk/pynni/nni/feature_engineering/gbdt_selector/gbdt_selector.py → nni/feature_engineering/gbdt_selector/gbdt_selector.py
View file @ f98ee672
...@@ -25,11 +25,12 @@ gbdt_selector.py including: ...@@ -25,11 +25,12 @@ gbdt_selector.py including:
import random import random
from sklearn.model_selection import train_test_split from sklearn.model_selection import train_test_split
from nni.feature_engineering.feature_selector import FeatureSelector
# pylint: disable=E0401 # pylint: disable=E0401
import lightgbm as lgb import lightgbm as lgb
from nni.feature_engineering.feature_selector import FeatureSelector
class GBDTSelector(FeatureSelector): class GBDTSelector(FeatureSelector):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment