modelfoxdotdev / modelfox

ModelFox makes it easy to train, deploy, and monitor machine learning models.
Other
1.46k stars 63 forks source link

More informative CLI train output #44

Closed otsaw closed 2 years ago

otsaw commented 2 years ago

Currently when training a model, the output I get is

Computing features. 0ms
Training model 2 of 8. 0s 116ms
Computing model comparison features. 0ms
Computing comparison metric. 0ms

I'd prefer to see what kind of model it's training and what the value of the comparison metric is. So, something similar to in the app's "All Models" table under "Grid", but save me a bit of time not having to go there. For example

Computing features. 0ms
Training model 2 of 8: Linear. 0s 116ms
Computing model comparison features. 0ms
Computing comparison metric: AUC 0.765832. 0ms
...
Finalizing the best model: Linear AUC 0.765832. 0ms

I also noticed that I can guess the final model type based on filesize, but it would be nice if it was clearly in the output.

isabella commented 2 years ago

Hi @otsaw! You are right we definitely should add this information to the CLI. I'll keep this issue open until we implement this.

deciduously commented 2 years ago

61 closes