Added the support of mlflow

[neogyk]

Provided the tracking of the experiment using the mlflow. It will store the train, test loss values, regularization and learning rate. Also, the config file parameters will be stored to the sqllite file.