kddcup runtime plots

chr-peters · Jun 7, 2021 · 46abe5d · 46abe5d
1 parent eaddf8a
commit 46abe5d
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/notebooks/runtime_plots.ipynb b/notebooks/runtime_plots.ipynb
@@ -7,7 +7,7 @@
  "outputs": [],
  "source": [
  "from sketching import settings\n",
- "from sketching.datasets import Dataset, Covertype_Sklearn\n",
+ "from sketching.datasets import Dataset, Covertype_Sklearn, KDDCup_Sklearn\n",
  "\n",
  "import pandas as pd\n",
  "\n",
@@ -76,6 +76,7 @@
  "\n",
  " titles = {\n",
  " \"covertype_sklearn\": \"Covertype\",\n",
+ " \"kddcup_sklearn\": \"Kddcup\"\n",
  " }\n",
  "\n",
  " for cur_method in [\"l2s\", \"uniform\", \"sketching\"]:\n",
@@ -116,6 +117,17 @@
  "make_plot(dataset, x_min=None, x_max=None, y_min=None, y_max=None, sampling_time=False)\n",
  "make_plot(dataset, x_min=None, x_max=None, y_min=None, y_max=None, sampling_time=True)"
  ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {},
+ "outputs": [],
+ "source": [
+ "dataset = KDDCup_Sklearn()\n",
+ "make_plot(dataset, x_min=None, x_max=None, y_min=1, y_max=10, sampling_time=False)\n",
+ "make_plot(dataset, x_min=None, x_max=None, y_min=1, y_max=6, sampling_time=True)"
+ ]
  }
  ],
  "metadata": {