diff --git a/examples/tutorials/try_optuna.ipynb b/examples/tutorials/try_optuna.ipynb
new file mode 100644
index 000000000..ed1871ad7
--- /dev/null
+++ b/examples/tutorials/try_optuna.ipynb
@@ -0,0 +1,1199 @@
+{
+ "cells": [
+ {
+ "cell_type": "code",
+ "execution_count": 2,
+ "id": "c25b63fd",
+ "metadata": {},
+ "outputs": [],
+ "source": [
+ "# --- Setup: Logging and Display Configuration ---\n",
+ "# Configure logging to see training progress and plotly to render as PNG for VS Code compatibility\n",
+ "import logging\n",
+ "import pandas as pd\n",
+ "import plotly.io as pio\n",
+ "\n",
+ "pd.options.plotting.backend = \"plotly\"\n",
+ "pio.renderers.default = \"png\" # Use PNG for VS Code notebook compatibility\n",
+ "\n",
+ "logging.basicConfig(level=logging.INFO, format=\"[%(asctime)s][%(levelname)s] %(message)s\")\n",
+ "logger = logging.getLogger(__name__)\n",
+ "logging.getLogger(\"choreographer\").setLevel(logging.ERROR)\n",
+ "logging.getLogger(\"kaleido\").setLevel(logging.ERROR)\n",
+ "logging.getLogger(\"choreographer\").disabled = True\n",
+ "logging.getLogger(\"kaleido\").disabled = True"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 3,
+ "id": "84299333",
+ "metadata": {},
+ "outputs": [
+ {
+ "name": "stdout",
+ "output_type": "stream",
+ "text": [
+ "Downloading load_measurements/mv_feeder/OS Gorredijk.parquet...\n"
+ ]
+ },
+ {
+ "name": "stderr",
+ "output_type": "stream",
+ "text": [
+ "/Users/fleur.petit/projects/openstef/.venv/lib/python3.12/site-packages/huggingface_hub/utils/_validators.py:202: UserWarning:\n",
+ "\n",
+ "The `local_dir_use_symlinks` argument is deprecated and ignored in `hf_hub_download`. Downloading to a local directory does not use symlinks anymore.\n",
+ "\n",
+ "[2026-03-05 16:50:42,135][INFO] HTTP Request: HEAD https://huggingface.co/datasets/OpenSTEF/liander2024-energy-forecasting-benchmark/resolve/main/load_measurements/mv_feeder/OS%20Gorredijk.parquet \"HTTP/1.1 302 Found\"\n",
+ "Warning: You are sending unauthenticated requests to the HF Hub. Please set a HF_TOKEN to enable higher rate limits and faster downloads.\n",
+ "[2026-03-05 16:50:42,137][WARNING] Warning: You are sending unauthenticated requests to the HF Hub. Please set a HF_TOKEN to enable higher rate limits and faster downloads.\n",
+ "[2026-03-05 16:50:42,275][INFO] HTTP Request: HEAD https://huggingface.co/datasets/OpenSTEF/liander2024-energy-forecasting-benchmark/resolve/main/weather_forecasts_versioned/mv_feeder/OS%20Gorredijk.parquet \"HTTP/1.1 302 Found\"\n"
+ ]
+ },
+ {
+ "name": "stdout",
+ "output_type": "stream",
+ "text": [
+ "✓ load_measurements/mv_feeder/OS Gorredijk.parquet downloaded\n",
+ "Downloading weather_forecasts_versioned/mv_feeder/OS Gorredijk.parquet...\n",
+ "✓ weather_forecasts_versioned/mv_feeder/OS Gorredijk.parquet downloaded\n",
+ "Downloading EPEX.parquet...\n"
+ ]
+ },
+ {
+ "name": "stderr",
+ "output_type": "stream",
+ "text": [
+ "[2026-03-05 16:50:42,572][INFO] HTTP Request: HEAD https://huggingface.co/datasets/OpenSTEF/liander2024-energy-forecasting-benchmark/resolve/main/EPEX.parquet \"HTTP/1.1 302 Found\"\n",
+ "[2026-03-05 16:50:42,701][INFO] HTTP Request: HEAD https://huggingface.co/datasets/OpenSTEF/liander2024-energy-forecasting-benchmark/resolve/main/profiles.parquet \"HTTP/1.1 302 Found\"\n"
+ ]
+ },
+ {
+ "name": "stdout",
+ "output_type": "stream",
+ "text": [
+ "✓ EPEX.parquet downloaded\n",
+ "Downloading profiles.parquet...\n",
+ "✓ profiles.parquet downloaded\n",
+ "\n",
+ "✅ All files downloaded successfully!\n"
+ ]
+ }
+ ],
+ "source": [
+ "# Download dataset from HuggingFace Hub\n",
+ "# The dataset is stored as parquet files for efficient loading\n",
+ "from huggingface_hub import hf_hub_download # pyright: ignore[reportUnknownVariableType]\n",
+ "from openstef_core.base_model import Path\n",
+ "\n",
+ "repo_id = \"OpenSTEF/liander2024-energy-forecasting-benchmark\" # Public benchmark dataset\n",
+ "local_dir = Path(\"./liander_dataset\")\n",
+ "target = \"mv_feeder/OS Gorredijk\" # Specific installation to focus on\n",
+ "\n",
+ "# Download required files: load measurements, weather, prices, and profiles\n",
+ "files_to_download = [\n",
+ " f\"load_measurements/{target}.parquet\", # Energy consumption data\n",
+ " f\"weather_forecasts_versioned/{target}.parquet\", # Weather features\n",
+ " \"EPEX.parquet\", # Electricity prices (optional feature)\n",
+ " \"profiles.parquet\" # Standard load profiles (optional feature)\n",
+ "]\n",
+ "\n",
+ "for filename in files_to_download:\n",
+ " print(f\"Downloading {filename}...\")\n",
+ " hf_hub_download(repo_id=repo_id, filename=filename, repo_type=\"dataset\",\n",
+ " local_dir=local_dir, local_dir_use_symlinks=False) # pyright: ignore[reportCallIssue]\n",
+ " print(f\"✓ {filename} downloaded\")\n",
+ "\n",
+ "print(\"\\n✅ All files downloaded successfully!\")"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 4,
+ "id": "524f65a1",
+ "metadata": {},
+ "outputs": [
+ {
+ "name": "stderr",
+ "output_type": "stream",
+ "text": [
+ "[2026-03-05 16:50:45,304][WARNING] Parquet file does not contain 'sample_interval' attribute. Using default value of 15 minutes.\n",
+ "[2026-03-05 16:50:45,336][WARNING] Parquet file does not contain 'sample_interval' attribute. Using default value of 15 minutes.\n",
+ "[2026-03-05 16:50:45,365][WARNING] Parquet file does not contain 'sample_interval' attribute. Using default value of 15 minutes.\n",
+ "[2026-03-05 16:50:45,378][WARNING] Parquet file does not contain 'sample_interval' attribute. Using default value of 15 minutes.\n"
+ ]
+ },
+ {
+ "name": "stdout",
+ "output_type": "stream",
+ "text": [
+ "Dataset shape: (35136, 28)\n",
+ "Date range: 2024-01-01 00:00:00+00:00 to 2024-12-31 23:45:00+00:00\n"
+ ]
+ },
+ {
+ "data": {
+ "application/vnd.microsoft.datawrangler.viewer.v0+json": {
+ "columns": [
+ {
+ "name": "timestamp",
+ "rawType": "datetime64[ns, UTC]",
+ "type": "unknown"
+ },
+ {
+ "name": "load",
+ "rawType": "float64",
+ "type": "float"
+ },
+ {
+ "name": "temperature_2m",
+ "rawType": "float32",
+ "type": "float"
+ },
+ {
+ "name": "relative_humidity_2m",
+ "rawType": "float32",
+ "type": "float"
+ },
+ {
+ "name": "surface_pressure",
+ "rawType": "float32",
+ "type": "float"
+ },
+ {
+ "name": "cloud_cover",
+ "rawType": "float32",
+ "type": "float"
+ },
+ {
+ "name": "wind_speed_10m",
+ "rawType": "float32",
+ "type": "float"
+ },
+ {
+ "name": "wind_speed_80m",
+ "rawType": "float32",
+ "type": "float"
+ },
+ {
+ "name": "wind_direction_10m",
+ "rawType": "float32",
+ "type": "float"
+ },
+ {
+ "name": "shortwave_radiation",
+ "rawType": "float32",
+ "type": "float"
+ },
+ {
+ "name": "direct_radiation",
+ "rawType": "float32",
+ "type": "float"
+ },
+ {
+ "name": "diffuse_radiation",
+ "rawType": "float32",
+ "type": "float"
+ },
+ {
+ "name": "direct_normal_irradiance",
+ "rawType": "float32",
+ "type": "float"
+ },
+ {
+ "name": "EPEX_NL",
+ "rawType": "float64",
+ "type": "float"
+ },
+ {
+ "name": "E1A_AZI_A",
+ "rawType": "float64",
+ "type": "float"
+ },
+ {
+ "name": "E1A_AMI_A",
+ "rawType": "float64",
+ "type": "float"
+ },
+ {
+ "name": "E1B_AZI_A",
+ "rawType": "float64",
+ "type": "float"
+ },
+ {
+ "name": "E1B_AMI_A",
+ "rawType": "float64",
+ "type": "float"
+ },
+ {
+ "name": "E1C_AZI_A",
+ "rawType": "float64",
+ "type": "float"
+ },
+ {
+ "name": "E1C_AMI_A",
+ "rawType": "float64",
+ "type": "float"
+ },
+ {
+ "name": "E2A_AZI_A",
+ "rawType": "float64",
+ "type": "float"
+ },
+ {
+ "name": "E2A_AMI_A",
+ "rawType": "float64",
+ "type": "float"
+ },
+ {
+ "name": "E2B_AZI_A",
+ "rawType": "float64",
+ "type": "float"
+ },
+ {
+ "name": "E2B_AMI_A",
+ "rawType": "float64",
+ "type": "float"
+ },
+ {
+ "name": "E3A_A",
+ "rawType": "float64",
+ "type": "float"
+ },
+ {
+ "name": "E3B_A",
+ "rawType": "float64",
+ "type": "float"
+ },
+ {
+ "name": "E3C_A",
+ "rawType": "float64",
+ "type": "float"
+ },
+ {
+ "name": "E3D_A",
+ "rawType": "float64",
+ "type": "float"
+ },
+ {
+ "name": "E4A_A",
+ "rawType": "float64",
+ "type": "float"
+ }
+ ],
+ "ref": "aa9bfad1-6c3e-4ed7-8923-65e22d67bcbf",
+ "rows": [
+ [
+ "2024-01-01 00:00:00+00:00",
+ "423333.3333333333",
+ "7.2435",
+ "85.02532",
+ "994.23645",
+ "100.0",
+ "28.185953",
+ "43.832863",
+ "204.92845",
+ "0.0",
+ "0.0",
+ "0.0",
+ "0.0",
+ "0.01",
+ "2.97e-05",
+ "4.031e-05",
+ "6.206e-05",
+ "7.815e-05",
+ "5.683e-05",
+ "6.431e-05",
+ "2.42e-05",
+ "3.404e-05",
+ "5.292e-05",
+ "6.425e-05",
+ "5.839e-05",
+ "5.839e-05",
+ "5.839e-05",
+ "5.839e-05",
+ "7.931e-05"
+ ],
+ [
+ "2024-01-01 00:15:00+00:00",
+ "436666.6666666666",
+ "7.281",
+ "84.80853",
+ "994.1865",
+ "100.0",
+ "28.75338",
+ "44.97622",
+ "206.93102",
+ "0.0",
+ "0.0",
+ "0.0",
+ "0.0",
+ "0.01",
+ "2.91e-05",
+ "3.866e-05",
+ "6.079e-05",
+ "7.537e-05",
+ "5.572e-05",
+ "6.058e-05",
+ "2.395e-05",
+ "3.352e-05",
+ "5.236e-05",
+ "6.326e-05",
+ "5.826e-05",
+ "5.826e-05",
+ "5.826e-05",
+ "5.826e-05",
+ "7.931e-05"
+ ],
+ [
+ "2024-01-01 00:30:00+00:00",
+ "410000.0",
+ "7.3185005",
+ "84.59174",
+ "994.1366",
+ "100.0",
+ "29.320807",
+ "46.11957",
+ "208.93356",
+ "0.0",
+ "0.0",
+ "0.0",
+ "0.0",
+ "0.01",
+ "2.794e-05",
+ "3.771e-05",
+ "5.799e-05",
+ "7.33e-05",
+ "5.451e-05",
+ "5.967e-05",
+ "2.338e-05",
+ "3.312e-05",
+ "5.113e-05",
+ "6.252e-05",
+ "5.782e-05",
+ "5.782e-05",
+ "5.782e-05",
+ "5.782e-05",
+ "7.931e-05"
+ ],
+ [
+ "2024-01-01 00:45:00+00:00",
+ "403333.3333333333",
+ "7.3560004",
+ "84.374954",
+ "994.08673",
+ "100.0",
+ "29.888233",
+ "47.262924",
+ "210.93613",
+ "0.0",
+ "0.0",
+ "0.0",
+ "0.0",
+ "0.01",
+ "2.712e-05",
+ "3.649e-05",
+ "5.659e-05",
+ "7.12e-05",
+ "5.211e-05",
+ "5.708e-05",
+ "2.325e-05",
+ "3.219e-05",
+ "5.083e-05",
+ "6.076e-05",
+ "5.89e-05",
+ "5.89e-05",
+ "5.89e-05",
+ "5.89e-05",
+ "7.931e-05"
+ ],
+ [
+ "2024-01-01 01:00:00+00:00",
+ "420000.0",
+ "7.3935003",
+ "84.158165",
+ "994.0368",
+ "100.0",
+ "30.45566",
+ "48.40628",
+ "212.93869",
+ "0.0",
+ "0.0",
+ "0.0",
+ "0.0",
+ "0.0",
+ "2.714e-05",
+ "3.44e-05",
+ "5.728e-05",
+ "6.668e-05",
+ "5.045e-05",
+ "5.493e-05",
+ "2.359e-05",
+ "3.198e-05",
+ "5.158e-05",
+ "6.036e-05",
+ "5.726e-05",
+ "5.726e-05",
+ "5.726e-05",
+ "5.726e-05",
+ "7.931e-05"
+ ]
+ ],
+ "shape": {
+ "columns": 28,
+ "rows": 5
+ }
+ },
+ "text/html": [
+ "\n",
+ "\n",
+ "
\n",
+ " \n",
+ " \n",
+ " | \n",
+ " load | \n",
+ " temperature_2m | \n",
+ " relative_humidity_2m | \n",
+ " surface_pressure | \n",
+ " cloud_cover | \n",
+ " wind_speed_10m | \n",
+ " wind_speed_80m | \n",
+ " wind_direction_10m | \n",
+ " shortwave_radiation | \n",
+ " direct_radiation | \n",
+ " ... | \n",
+ " E1C_AMI_A | \n",
+ " E2A_AZI_A | \n",
+ " E2A_AMI_A | \n",
+ " E2B_AZI_A | \n",
+ " E2B_AMI_A | \n",
+ " E3A_A | \n",
+ " E3B_A | \n",
+ " E3C_A | \n",
+ " E3D_A | \n",
+ " E4A_A | \n",
+ "
\n",
+ " \n",
+ " | timestamp | \n",
+ " | \n",
+ " | \n",
+ " | \n",
+ " | \n",
+ " | \n",
+ " | \n",
+ " | \n",
+ " | \n",
+ " | \n",
+ " | \n",
+ " | \n",
+ " | \n",
+ " | \n",
+ " | \n",
+ " | \n",
+ " | \n",
+ " | \n",
+ " | \n",
+ " | \n",
+ " | \n",
+ " | \n",
+ "
\n",
+ " \n",
+ " \n",
+ " \n",
+ " | 2024-01-01 00:00:00+00:00 | \n",
+ " 423333.333333 | \n",
+ " 7.243500 | \n",
+ " 85.025322 | \n",
+ " 994.236450 | \n",
+ " 100.0 | \n",
+ " 28.185953 | \n",
+ " 43.832863 | \n",
+ " 204.928452 | \n",
+ " 0.0 | \n",
+ " 0.0 | \n",
+ " ... | \n",
+ " 0.000064 | \n",
+ " 0.000024 | \n",
+ " 0.000034 | \n",
+ " 0.000053 | \n",
+ " 0.000064 | \n",
+ " 0.000058 | \n",
+ " 0.000058 | \n",
+ " 0.000058 | \n",
+ " 0.000058 | \n",
+ " 0.000079 | \n",
+ "
\n",
+ " \n",
+ " | 2024-01-01 00:15:00+00:00 | \n",
+ " 436666.666667 | \n",
+ " 7.281000 | \n",
+ " 84.808533 | \n",
+ " 994.186523 | \n",
+ " 100.0 | \n",
+ " 28.753380 | \n",
+ " 44.976219 | \n",
+ " 206.931015 | \n",
+ " 0.0 | \n",
+ " 0.0 | \n",
+ " ... | \n",
+ " 0.000061 | \n",
+ " 0.000024 | \n",
+ " 0.000034 | \n",
+ " 0.000052 | \n",
+ " 0.000063 | \n",
+ " 0.000058 | \n",
+ " 0.000058 | \n",
+ " 0.000058 | \n",
+ " 0.000058 | \n",
+ " 0.000079 | \n",
+ "
\n",
+ " \n",
+ " | 2024-01-01 00:30:00+00:00 | \n",
+ " 410000.000000 | \n",
+ " 7.318501 | \n",
+ " 84.591743 | \n",
+ " 994.136597 | \n",
+ " 100.0 | \n",
+ " 29.320807 | \n",
+ " 46.119572 | \n",
+ " 208.933563 | \n",
+ " 0.0 | \n",
+ " 0.0 | \n",
+ " ... | \n",
+ " 0.000060 | \n",
+ " 0.000023 | \n",
+ " 0.000033 | \n",
+ " 0.000051 | \n",
+ " 0.000063 | \n",
+ " 0.000058 | \n",
+ " 0.000058 | \n",
+ " 0.000058 | \n",
+ " 0.000058 | \n",
+ " 0.000079 | \n",
+ "
\n",
+ " \n",
+ " | 2024-01-01 00:45:00+00:00 | \n",
+ " 403333.333333 | \n",
+ " 7.356000 | \n",
+ " 84.374954 | \n",
+ " 994.086731 | \n",
+ " 100.0 | \n",
+ " 29.888233 | \n",
+ " 47.262924 | \n",
+ " 210.936127 | \n",
+ " 0.0 | \n",
+ " 0.0 | \n",
+ " ... | \n",
+ " 0.000057 | \n",
+ " 0.000023 | \n",
+ " 0.000032 | \n",
+ " 0.000051 | \n",
+ " 0.000061 | \n",
+ " 0.000059 | \n",
+ " 0.000059 | \n",
+ " 0.000059 | \n",
+ " 0.000059 | \n",
+ " 0.000079 | \n",
+ "
\n",
+ " \n",
+ " | 2024-01-01 01:00:00+00:00 | \n",
+ " 420000.000000 | \n",
+ " 7.393500 | \n",
+ " 84.158165 | \n",
+ " 994.036804 | \n",
+ " 100.0 | \n",
+ " 30.455660 | \n",
+ " 48.406281 | \n",
+ " 212.938690 | \n",
+ " 0.0 | \n",
+ " 0.0 | \n",
+ " ... | \n",
+ " 0.000055 | \n",
+ " 0.000024 | \n",
+ " 0.000032 | \n",
+ " 0.000052 | \n",
+ " 0.000060 | \n",
+ " 0.000057 | \n",
+ " 0.000057 | \n",
+ " 0.000057 | \n",
+ " 0.000057 | \n",
+ " 0.000079 | \n",
+ "
\n",
+ " \n",
+ "
\n",
+ "
5 rows × 28 columns
\n",
+ "