Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

AutoTSEstimator to support customized loss function and metric #4035

Open
shupingyeh opened this issue Feb 16, 2022 · 4 comments · Fixed by #4063
Open

AutoTSEstimator to support customized loss function and metric #4035

shupingyeh opened this issue Feb 16, 2022 · 4 comments · Fixed by #4063
Assignees

Comments

@shupingyeh
Copy link

Hi,

Can AutoTSEstimator support customized loss function and metric for AutoML tuning?

@TheaperDeng
Copy link
Collaborator

Customized loss is supported. https://bigdl.readthedocs.io/en/latest/doc/PythonAPI/Chronos/autotsestimator.html.
String, pytorch/tf.keras loss instance or pytorch loss creator function are all supported

If customized metric is really needed we can add the support.

@TheaperDeng
Copy link
Collaborator

The latest nightly build has supported this feature (customized metric). https://pypi.org/project/bigdl-chronos/2.1.0b20220228/

@liangs6212 could you have some thorough UTs on this?

@shupingyeh
Copy link
Author

Thank you very much!

@liangs6212
Copy link
Contributor

The latest nightly build has supported this feature (customized metric). https://pypi.org/project/bigdl-chronos/2.1.0b20220228/

@liangs6212 could you have some thorough UTs on this?

ok, I'll add some relevant UTs.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
4 participants