[ML][Training] Model saving to file