aksnzhy / xlearn

High performance, easy-to-use, and scalable machine learning (ML) package, including linear model (LR), factorization machines (FM), and field-aware factorization machines (FFM) for Python and CLI interface.
https://xlearn-doc.readthedocs.io/en/latest/index.html
Apache License 2.0
3.09k stars 519 forks source link

How to Extract Training Loss versus Epoch #372

Open tinnguyen96 opened 1 year ago

tinnguyen96 commented 1 year ago

Hi! I would like to programmatically extract the train mse_loss incurred after each training epoch. Currently, I am piping the xlearn printing output to a file, but I don't know how to extract the mse_loss from this .txt log. Is there a more straightforward way to save train mse_loss using either the sklearn API or the native Python API? Thank you!