Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel
Jeff Nirschl 63bd95a666
Re-configure stage train_model to put metrics.json under Git version control instead of putting it in the dvc cache
3 years ago
..
9557af78d7
Update script train_model.py to accept training dataset and pre-defined cross-validation splits, train an SK-Learn estimator (currently RF), and output a pickle file of the CV estimator and performance metrics. Also, built a custom function to compute the geometric mean of precision and recall. Re-configured params.yaml to have a base variable "classifier" that defines the current classifier in use and a separate dict "model_params" that includes sub-dicts for the various model-specific parameters. DVC running successfully.
3 years ago
63bd95a666
Re-configure stage train_model to put metrics.json under Git version control instead of putting it in the dvc cache
3 years ago

Comments

Loading...