Skip to content

Commit

Permalink
Add initial version of auto estimator (intel-analytics#3731)
Browse files Browse the repository at this point in the history
* add initial version of auto estimator

* support str for torch optimizor and loss

* move create searcher

* change name to model_builder

* add best model

* add convert probability to class in metrics.Accuracy

* change condition order

* add pytorch ut

* change util name

* add LR_NAME

* add document for LR_NAME

* add check with best model and optimizer class test

* add ut for tf.keras

* move tests to orca/automl/autoestimator/

* change name

* add auto test

* fix pep8

* remove return in fit

* add raise error for fit multiple times

* remove optimizer class

* fix bug

* change error message
  • Loading branch information
shanyu-sys committed Apr 13, 2021
1 parent 9a0fe89 commit caf87a7
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions python/orca/src/test/dev/run-pytests-ray
Original file line number Diff line number Diff line change
Expand Up @@ -79,3 +79,13 @@ then
fi

ray stop -f

echo "Running orca auto estimator tests"
python -m pytest -v ../test/zoo/orca/automl/autoestimator
exit_status_6=$?
if [ $exit_status_6 -ne 0 ];
then
exit $exit_status_6
fi

ray stop -f

0 comments on commit caf87a7

Please sign in to comment.