-
Notifications
You must be signed in to change notification settings - Fork 39
Closed
Labels
A: frameworksArea: ML Framework integrationArea: ML Framework integrationfeature requestgood first issue
Description
Extend the existing XGBoost integration (dvc.xgb) to include model saving functionality. Extend test_xgboost.py to cover this feature.
Ref:
- Official XGBoost docs on saving models: https://xgboost.readthedocs.io/en/stable/tutorials/saving_model.html & https://mljar.com/blog/xgboost-save-load-python/
- Similar extension added to
kerasintegration: Save checkpoint for Keras #69
Metadata
Metadata
Assignees
Labels
A: frameworksArea: ML Framework integrationArea: ML Framework integrationfeature requestgood first issue