From de76aa4d57994f2b762542f684962aab35729f14 Mon Sep 17 00:00:00 2001 From: Jianbo Date: Tue, 19 Nov 2019 20:30:13 -0600 Subject: [PATCH] refactor(examples/trails/sklearn)-remove empty lines --- examples/trials/sklearn/classification/main.py | 5 ----- examples/trials/sklearn/regression/main.py | 5 ----- 2 files changed, 10 deletions(-) diff --git a/examples/trials/sklearn/classification/main.py b/examples/trials/sklearn/classification/main.py index 7b320ec6c0..6839e830f6 100644 --- a/examples/trials/sklearn/classification/main.py +++ b/examples/trials/sklearn/classification/main.py @@ -25,7 +25,6 @@ LOG = logging.getLogger('sklearn_classification') - def load_data(): '''Load dataset, use 20newsgroups dataset''' digits = load_digits() @@ -38,7 +37,6 @@ def load_data(): return X_train, X_test, y_train, y_test - def get_default_parameters(): '''get default parameters''' params = { @@ -50,7 +48,6 @@ def get_default_parameters(): } return params - def get_model(PARAMS): '''Get model according to parameters''' model = SVC() @@ -62,7 +59,6 @@ def get_model(PARAMS): return model - def run(X_train, X_test, y_train, y_test, model): '''Train model and predict result''' model.fit(X_train, y_train) @@ -70,7 +66,6 @@ def run(X_train, X_test, y_train, y_test, model): LOG.debug('score: %s' % score) nni.report_final_result(score) - if __name__ == '__main__': X_train, X_test, y_train, y_test = load_data() diff --git a/examples/trials/sklearn/regression/main.py b/examples/trials/sklearn/regression/main.py index 964dc499d5..af54bf225f 100644 --- a/examples/trials/sklearn/regression/main.py +++ b/examples/trials/sklearn/regression/main.py @@ -30,7 +30,6 @@ LOG = logging.getLogger('sklearn_regression') - def load_data(): '''Load dataset, use boston dataset''' boston = load_boston() @@ -47,13 +46,11 @@ def load_data(): return X_train, X_test, y_train, y_test - def get_default_parameters(): '''get default parameters''' params = {'model_name': 'LinearRegression'} return params - def get_model(PARAMS): '''Get model according to parameters''' model_dict = { @@ -78,7 +75,6 @@ def get_model(PARAMS): raise return model - def run(X_train, X_test, y_train, y_test, model): '''Train model and predict result''' model.fit(X_train, y_train) @@ -87,7 +83,6 @@ def run(X_train, X_test, y_train, y_test, model): LOG.debug('r2 score: %s' % score) nni.report_final_result(score) - if __name__ == '__main__': X_train, X_test, y_train, y_test = load_data()