update import path for feature engineering examples (#3274)

This commit is contained in:
98may 2021-01-07 21:35:54 +08:00 коммит произвёл GitHub
Родитель 8657ef339d
Коммит 6e5584a8ec
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
3 изменённых файлов: 4 добавлений и 4 удалений

Просмотреть файл

@ -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()