Infer `num_classes` for the label using vocab

google / nitroml

NitroML is a modular, portable, and scalable model-quality benchmarking framework for Machine Learning and Automated Machine Learning (AutoML) pipelines.

Apache License 2.0

42 stars 6 forks source link

Infer `num_classes` for the label using vocab #28

Closed nikhil-dce closed 4 years ago

nikhil-dce commented 4 years ago

While creating the task object, infer num_classes using the size of the vocab after preprocessing.

cweill commented 4 years ago

Fixed with 1b1f12614af6c88965344f070b6bab7b66925357