diff --git a/notebooks/runtime_plots.ipynb b/notebooks/runtime_plots.ipynb index 3a8a43c..0547666 100644 --- 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": {