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

add h2o schema #413

Merged
merged 8 commits into from
Mar 5, 2024
Merged

add h2o schema #413

merged 8 commits into from
Mar 5, 2024

Conversation

ryanSoley
Copy link
Member

@ryanSoley ryanSoley commented Feb 26, 2024

What

  • adds the ability for self artifacts to be logged with Experiment methods other than log_artifact
  • adding schema for some h2o models
    • H2OGeneralizedLinearEstimator
    • H2OGradientBoostingEstimator
    • H2ORandomForestEstimator
    • H2OTargetEncoderEstimator
    • H2OXGBoostEstimator

How to Test

  • python -m pytest tests/integration/test_schema.py

@ryanSoley ryanSoley self-assigned this Feb 26, 2024
@ryanSoley ryanSoley added the enhancement Issues & PRs that capture enhancements to existing features. Issues require associated PRs. label Feb 26, 2024
@ryanSoley ryanSoley force-pushed the h2o-schema branch 2 times, most recently from 4aa1454 to 55615d8 Compare February 27, 2024 20:01
@ryanSoley
Copy link
Member Author

needs #415 merged to log the model artifacts via schema

@ryanSoley ryanSoley marked this pull request as ready for review February 29, 2024 17:07
@ryanSoley ryanSoley requested a review from a team as a code owner February 29, 2024 17:07
@ryanSoley
Copy link
Member Author

ryanSoley commented Feb 29, 2024

the h2o model contains a metrics table and feature importances, but they are stored in h2o specific data types. I'll open followup stories to investigate how to get that data logged with the schema too. for now, its all available on the estimator logged with log_h2o_model

@ryanSoley ryanSoley merged commit 02e9842 into main Mar 5, 2024
7 checks passed
@ryanSoley ryanSoley deleted the h2o-schema branch March 5, 2024 15:51
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement Issues & PRs that capture enhancements to existing features. Issues require associated PRs.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants