Skip to content

Commit

Permalink
hyperopt update
Browse files Browse the repository at this point in the history
  • Loading branch information
lewis-morris committed Jan 27, 2020
1 parent 95d1fdf commit edc7b8e
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 3 deletions.
4 changes: 2 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,15 +7,15 @@

setup(
name='Skperopt',
version='0.0.695',
version='0.0.7',
packages=["skperopt"],
url='https://github.com/lewis-morris/Skperopt',
license='MIT',
author='Lewis',
author_email='[email protected]',
description='Hyperopt Wrapper',
install_requires=requirements,
download_url = 'https://github.com/lewis-morris/Skperopt/archive/0.0.695.tar.gz',
download_url = 'https://github.com/lewis-morris/Skperopt/archive/0.0.7.tar.gz',
keywords=['hyperopt-wrapper', 'hyperparameter'],
classifiers=[
'Development Status :: 5 - Production/Stable',
Expand Down
4 changes: 3 additions & 1 deletion skperopt/search.py
Original file line number Diff line number Diff line change
Expand Up @@ -113,7 +113,7 @@ def scorer_is_better(test_type, new_score, old_score):
return True
else:
return False
elif test_type == "rmse":
else:
if old_score >= new_score:
return True
else:
Expand All @@ -139,6 +139,8 @@ def get_score(y_true, y_pred, scorer):
score_list.append(f1_score(y_true, y_pred, average='macro'))
elif score_type == "rmse":
score_list.append(mean_squared_error(y_true,y_pred,squared = False))
elif score_type == "mse":
score_list.append(mean_squared_error(y_true, y_pred, squared=True))
elif score_type == "auc":
score_list.append(roc_auc_score(y_true, y_pred, average="macro"))
elif score_type == "accuracy":
Expand Down

0 comments on commit edc7b8e

Please sign in to comment.