Add MLflow logging and use .predict
method from Lightning
#952
This job was skipped
Loading
.predict
method from Lightning
#952