Support mlflow logging for line training and properly document it

MLflow support was only implemented for document training