fix (#46)

* timer

* plotROC bug

* add run_name

Author: Epstein

splicemachine/mlflow_support/mlflow_support.py

@@ -165,6 +165,8 @@ def _start_run(run_id=None, tags=None, experiment_id=None, run_name=None, nested
 
     for key in tags:
         mlflow.set_tag(key, tags[key])
+    if run_name:
+        mlflow.set_tag('mlflow.runName', run_name)
 
 
 @_mlflow_patch('log_pipeline_stages')
@@ -246,6 +248,7 @@ def _timer(timer_name, param=True):
         (mlflow.log_param if param else mlflow.log_metric)(timer_name, t1)
         print(
             f"Code Block {timer_name}:\nRan in {round(t1, 3)} secs\nRan in {round(t1 / 60, 3)} mins"
+        )
 
 
 @_mlflow_patch('download_artifact')

splicemachine/stats.py

@@ -171,8 +171,8 @@ def plotROC(self, fittedEstimator, ax):
             trainingSummary = fittedEstimator.summary
             roc = trainingSummary.roc.toPandas()
             ax.plot(roc['FPR'], roc['TPR'], label='Training set areaUnderROC: \n' + str(trainingSummary.areaUnderROC))
-            ax.set_ylabel('False Positive Rate')
-            ax.set_xlabel('True Positive Rate')
+            ax.set_xlabel('False Positive Rate')
+            ax.set_ylabel('True Positive Rate')
             ax.set_title('ROC Curve')
             ax.legend()
             return ax