Skip to content

Commit

Permalink
refactor(examples/trails/sklearn)-remove empty lines
Browse files Browse the repository at this point in the history
  • Loading branch information
JChen502 committed Nov 20, 2019
1 parent 823369f commit de76aa4
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 10 deletions.
5 changes: 0 additions & 5 deletions examples/trials/sklearn/classification/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,6 @@

LOG = logging.getLogger('sklearn_classification')


def load_data():
'''Load dataset, use 20newsgroups dataset'''
digits = load_digits()
Expand All @@ -38,7 +37,6 @@ def load_data():

return X_train, X_test, y_train, y_test


def get_default_parameters():
'''get default parameters'''
params = {
Expand All @@ -50,7 +48,6 @@ def get_default_parameters():
}
return params


def get_model(PARAMS):
'''Get model according to parameters'''
model = SVC()
Expand All @@ -62,15 +59,13 @@ 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)
score = model.score(X_test, y_test)
LOG.debug('score: %s' % score)
nni.report_final_result(score)


if __name__ == '__main__':
X_train, X_test, y_train, y_test = load_data()

Expand Down
5 changes: 0 additions & 5 deletions examples/trials/sklearn/regression/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,6 @@

LOG = logging.getLogger('sklearn_regression')


def load_data():
'''Load dataset, use boston dataset'''
boston = load_boston()
Expand All @@ -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 = {
Expand All @@ -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)
Expand All @@ -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()

Expand Down

0 comments on commit de76aa4

Please sign in to comment.