|
176 | 176 | "id": "2fd54c6c-3d28-4864-af38-3b22875d0f0a", |
177 | 177 | "metadata": {}, |
178 | 178 | "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." |
180 | 180 | ] |
181 | 181 | }, |
182 | 182 | { |
183 | 183 | "cell_type": "code", |
184 | | - "execution_count": 5, |
| 184 | + "execution_count": null, |
185 | 185 | "id": "b350b56f-5daa-411f-9090-b544760c34ef", |
186 | 186 | "metadata": {}, |
187 | 187 | "outputs": [], |
|
196 | 196 | "req_params = {'k': 3, 'G': 3, 'grid_e': 0.02}\n", |
197 | 197 | "\n", |
198 | 198 | "model = KAN(layer_dims = layer_dims,\n", |
199 | | - " layer_type = 'base',\n", |
| 199 | + " layer_type = 'spline',\n", |
200 | 200 | " required_parameters = req_params,\n", |
201 | 201 | " seed = seed\n", |
202 | 202 | " )" |
|
431 | 431 | }, |
432 | 432 | { |
433 | 433 | "cell_type": "code", |
434 | | - "execution_count": 11, |
| 434 | + "execution_count": null, |
435 | 435 | "id": "815a1407-04e4-49a0-a3d5-ae8722a65e86", |
436 | 436 | "metadata": {}, |
437 | 437 | "outputs": [], |
438 | 438 | "source": [ |
439 | 439 | "# Re-initialize the model\n", |
440 | 440 | "model = KAN(layer_dims = layer_dims,\n", |
441 | | - " layer_type = 'base',\n", |
| 441 | + " layer_type = 'spline',\n", |
442 | 442 | " required_parameters = req_params,\n", |
443 | 443 | " seed = seed\n", |
444 | 444 | " )\n", |
|
0 commit comments