Skip to content

Commit 96a345f

Browse files
committed
Developing DS Code
1 parent aef9c28 commit 96a345f

File tree

1 file changed

+27
-0
lines changed

1 file changed

+27
-0
lines changed
Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,27 @@
1+
# Databricks notebook source
2+
3+
4+
import mlflow
5+
from mlflow.tracking import MlflowClient
6+
7+
mlflow_client = MlflowClient()
8+
experiment = mlflow_client.get_experiment_by_name("/Shared/ciaran_experiment_nyc_taxi")
9+
experiment_id = experiment.experiment_id
10+
11+
12+
df = mlflow.search_runs(
13+
experiment_ids=experiment_id
14+
)
15+
16+
display(df)
17+
18+
df = df.rename(columns={"metrics.r2": "r2"})
19+
display(df)
20+
df = df[df.end_time.notnull()]
21+
df = df[df.r2.notnull()]
22+
23+
display(df)
24+
25+
df2 = df.drop(df[df['status'] == "FINISHED"].index, inplace = True)
26+
27+
display(df2)

0 commit comments

Comments
 (0)