Skip to content

Commit

Permalink
Small language tweaks to benchmark notebook (#152)
Browse files Browse the repository at this point in the history
mikeknep authored Aug 2, 2023
1 parent 11ab6c2 commit ecfef78
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions notebooks/benchmark.ipynb
Original file line number Diff line number Diff line change
@@ -115,8 +115,8 @@
"datasets = repo.list_datasets() # selects all Benchmark datasets\n",
"\n",
"# Other sample commands\n",
"# datasets = repo.list_datasets(datatype=\"time_series\") # select all time-series datasets\n",
"# datasets = repo.list_datasets(datatype=\"tabular\", tags=[\"small\", \"marketing\"]) # select all tabular_mixed, size small, and marketing-related datasets\n",
"# datasets = repo.list_datasets(datatype=\"tabular\") # select all tabular datasets\n",
"# datasets = repo.list_datasets(datatype=\"tabular\", tags=[\"small\", \"marketing\"]) # select all tabular, small, marketing-related datasets\n",
"\n",
"# This will show you all the datasets in the Benchmark dataset bucket\n",
"[dataset.name for dataset in datasets]"
@@ -268,7 +268,7 @@
"metadata": {},
"outputs": [],
"source": [
"comparison = b.compare(datasets=[heart_disease, iris], models=[GretelLSTM, GretelAmplify])"
"session = b.compare(datasets=[heart_disease, iris], models=[GretelLSTM, GretelAmplify])"
]
},
{
@@ -278,7 +278,7 @@
"outputs": [],
"source": [
"# Run this to see a snapshot of results! (While comparison is running)\n",
"comparison.results"
"session.results"
]
},
{
@@ -288,8 +288,8 @@
"outputs": [],
"source": [
"# Run this to wait for comparison to finish running, and export results as CSV at the end\n",
"comparison.wait()\n",
"comparison.export_results(\"./results.csv\")"
"session.wait()\n",
"session.export_results(\"./results.csv\")"
]
}
],

0 comments on commit ecfef78

Please sign in to comment.