"host/vscode:/vscode.git/clone" did not exist on "bd27ed6c38f3e4fd2d32bc84630fbeab2fe80d76"
Unverified Commit 6e5584a8 authored by 98may's avatar 98may Committed by GitHub
Browse files

update import path for feature engineering examples (#3274)

parent 8657ef33
...@@ -22,7 +22,7 @@ import numpy as np ...@@ -22,7 +22,7 @@ import numpy as np
from sklearn.datasets import load_svmlight_file from sklearn.datasets import load_svmlight_file
from sklearn.model_selection import train_test_split 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' 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') urllib.request.urlretrieve(url_zip_train, filename='train.bz2')
......
...@@ -33,7 +33,7 @@ from sklearn.linear_model import LogisticRegression ...@@ -33,7 +33,7 @@ from sklearn.linear_model import LogisticRegression
from sklearn.ensemble import ExtraTreesClassifier from sklearn.ensemble import ExtraTreesClassifier
from sklearn.feature_selection import SelectFromModel 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(): class Benchmark():
......
...@@ -28,7 +28,7 @@ from sklearn.linear_model import LogisticRegression ...@@ -28,7 +28,7 @@ from sklearn.linear_model import LogisticRegression
from sklearn.ensemble import ExtraTreesClassifier from sklearn.ensemble import ExtraTreesClassifier
from sklearn.feature_selection import SelectFromModel 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(): def test():
...@@ -54,4 +54,4 @@ def test(): ...@@ -54,4 +54,4 @@ def test():
print("Pipeline Score: ", pipeline.score(X_train, y_train)) print("Pipeline Score: ", pipeline.score(X_train, y_train))
if __name__ == "__main__": if __name__ == "__main__":
test() test()
\ No newline at end of file
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