aeon

patrickzib · May 8, 2023 · 63633ee · 63633ee
1 parent 73be897
commit 63633ee
Show file tree

Hide file tree

Showing 2 changed files with 173 additions and 40 deletions.
diff --git a/notebooks/classification.ipynb b/notebooks/classification.ipynb
@@ -3,17 +3,25 @@
   {
    "cell_type": "markdown",
    "id": "554f23a6",
-   "metadata": {},
+   "metadata": {
+    "pycharm": {
+     "name": "#%% md\n"
+    }
+   },
    "source": [
     "# Univariate time series classification with WEASEL 2.0\n",
     "\n",
-    "WEASEL 2.0 is build upon the sktime/sklearn-APIs."
+    "WEASEL 2.0 is build upon the aeon/sklearn-APIs."
    ]
   },
   {
    "cell_type": "markdown",
    "id": "4abb5608",
-   "metadata": {},
+   "metadata": {
+    "pycharm": {
+     "name": "#%% md\n"
+    }
+   },
    "source": [
     "First, we import the requires libraries"
    ]
@@ -22,19 +30,27 @@
    "cell_type": "code",
    "execution_count": 4,
    "id": "323d16ec",
-   "metadata": {},
+   "metadata": {
+    "pycharm": {
+     "name": "#%%\n"
+    }
+   },
    "outputs": [],
    "source": [
     "import time\n",
     "import numpy as np\n",
-    "from sktime.datasets import load_arrow_head\n",
+    "from aeon.datasets import load_arrow_head\n",
     "from weasel.classification.dictionary_based import WEASEL_V2"
    ]
   },
   {
    "cell_type": "markdown",
    "id": "3dea8779",
-   "metadata": {},
+   "metadata": {
+    "pycharm": {
+     "name": "#%% md\n"
+    }
+   },
    "source": [
     "Next, we load the arrow-head dataset"
    ]
@@ -43,7 +59,11 @@
    "cell_type": "code",
    "execution_count": 2,
    "id": "307b08a3",
-   "metadata": {},
+   "metadata": {
+    "pycharm": {
+     "name": "#%%\n"
+    }
+   },
    "outputs": [],
    "source": [
     "X_train, y_train = load_arrow_head(split=\"train\", return_type=\"numpy3d\")\n",
@@ -53,7 +73,11 @@
   {
    "cell_type": "markdown",
    "id": "e39e62b0",
-   "metadata": {},
+   "metadata": {
+    "pycharm": {
+     "name": "#%% md\n"
+    }
+   },
    "source": [
     "Finally, we fit a WEASEL 2.0 model and do some predictions"
    ]
@@ -62,7 +86,11 @@
    "cell_type": "code",
    "execution_count": 3,
    "id": "e7b739ac",
-   "metadata": {},
+   "metadata": {
+    "pycharm": {
+     "name": "#%%\n"
+    }
+   },
    "outputs": [
     {
      "name": "stdout",
@@ -97,7 +125,11 @@
    "cell_type": "code",
    "execution_count": null,
    "id": "7fd7c935",
-   "metadata": {},
+   "metadata": {
+    "pycharm": {
+     "name": "#%%\n"
+    }
+   },
    "outputs": [],
    "source": []
   }
@@ -123,4 +155,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 5
-}
+}