Skip to content

Commit 1dc84e0

Browse files
committed
Completely removed BaseLayer; no reason to keep it even for historical value
1 parent dd19cf5 commit 1dc84e0

File tree

10 files changed

+41
-909
lines changed

10 files changed

+41
-909
lines changed

docs/tutorials/Tutorial 1 - Getting Started.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,7 @@
178178
"Selecting and initializing a jaxKAN model is straightforward. Here we focus on the most basic model, `KAN`, where one needs to define the following arguments, which are universal for all KAN Layer types:\n",
179179
"\n",
180180
"- `layer_dims`: The dimensions for each of the network's layers' input/output nodes.\n",
181-
"- `layer_type`: The type of KAN layer to use, which defaults to `base` (more details can be found in the next tutorial for other layer types).\n",
181+
"- `layer_type`: The type of KAN layer to use, which defaults to `spline` (more details can be found in the next tutorial for other layer types).\n",
182182
"- `seed`: An integer for reproducibility.\n",
183183
"\n",
184184
"Apart from these, there is also the `required_parameters` argument, which is simply a dictionary with keys that correspond to additional parameters of the specific chosen layer.\n",

docs/tutorials/Tutorial 2 - Layer Types.ipynb

Lines changed: 18 additions & 88 deletions
Large diffs are not rendered by default.

docs/tutorials/Tutorial 7 - Adaptive State Transition.ipynb

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -176,12 +176,12 @@
176176
"id": "2fd54c6c-3d28-4864-af38-3b22875d0f0a",
177177
"metadata": {},
178178
"source": [
179-
"To showcase the adaptive state transition, we will use the Base KAN Layer, which corresponds to the original implementation of KANs using B-Splines. For more details about the required parameters of this type of layer, refer to the API Documentation."
179+
"To showcase the adaptive state transition, we will use the Spline KAN Layer, which corresponds to the original implementation of KANs using B-Splines. For more details about the required parameters of this type of layer, refer to the API Documentation."
180180
]
181181
},
182182
{
183183
"cell_type": "code",
184-
"execution_count": 5,
184+
"execution_count": null,
185185
"id": "b350b56f-5daa-411f-9090-b544760c34ef",
186186
"metadata": {},
187187
"outputs": [],
@@ -196,7 +196,7 @@
196196
"req_params = {'k': 3, 'G': 3, 'grid_e': 0.02}\n",
197197
"\n",
198198
"model = KAN(layer_dims = layer_dims,\n",
199-
" layer_type = 'base',\n",
199+
" layer_type = 'spline',\n",
200200
" required_parameters = req_params,\n",
201201
" seed = seed\n",
202202
" )"
@@ -431,14 +431,14 @@
431431
},
432432
{
433433
"cell_type": "code",
434-
"execution_count": 11,
434+
"execution_count": null,
435435
"id": "815a1407-04e4-49a0-a3d5-ae8722a65e86",
436436
"metadata": {},
437437
"outputs": [],
438438
"source": [
439439
"# Re-initialize the model\n",
440440
"model = KAN(layer_dims = layer_dims,\n",
441-
" layer_type = 'base',\n",
441+
" layer_type = 'spline',\n",
442442
" required_parameters = req_params,\n",
443443
" seed = seed\n",
444444
" )\n",

jaxkan/grids/BaseGrid.py

Lines changed: 0 additions & 154 deletions
This file was deleted.

0 commit comments

Comments
 (0)