зеркало из https://github.com/microsoft/nni.git
update import path for feature engineering examples (#3274)
This commit is contained in:
Родитель
8657ef339d
Коммит
6e5584a8ec
|
@ -22,7 +22,7 @@ import numpy as np
|
|||
from sklearn.datasets import load_svmlight_file
|
||||
from sklearn.model_selection import train_test_split
|
||||
|
||||
from nni.feature_engineering.gbdt_selector import GBDTSelector
|
||||
from nni.algorithms.feature_engineering.gbdt_selector import GBDTSelector
|
||||
|
||||
url_zip_train = 'https://www.csie.ntu.edu.tw/~cjlin/libsvmtools/datasets/binary/rcv1_train.binary.bz2'
|
||||
urllib.request.urlretrieve(url_zip_train, filename='train.bz2')
|
||||
|
|
|
@ -33,7 +33,7 @@ from sklearn.linear_model import LogisticRegression
|
|||
from sklearn.ensemble import ExtraTreesClassifier
|
||||
from sklearn.feature_selection import SelectFromModel
|
||||
|
||||
from nni.feature_engineering.gradient_selector import FeatureGradientSelector
|
||||
from nni.algorithms.feature_engineering.gradient_selector import FeatureGradientSelector
|
||||
|
||||
|
||||
class Benchmark():
|
||||
|
|
|
@ -28,7 +28,7 @@ from sklearn.linear_model import LogisticRegression
|
|||
from sklearn.ensemble import ExtraTreesClassifier
|
||||
from sklearn.feature_selection import SelectFromModel
|
||||
|
||||
from nni.feature_engineering.gradient_selector import FeatureGradientSelector
|
||||
from nni.algorithms.feature_engineering.gradient_selector import FeatureGradientSelector
|
||||
|
||||
|
||||
def test():
|
||||
|
@ -54,4 +54,4 @@ def test():
|
|||
print("Pipeline Score: ", pipeline.score(X_train, y_train))
|
||||
|
||||
if __name__ == "__main__":
|
||||
test()
|
||||
test()
|
||||
|
|
Загрузка…
Ссылка в новой задаче