{ "cells": [ { "cell_type": "code", "execution_count": 33, "id": "aa25df96", "metadata": {}, "outputs": [], "source": [ "import numpy as np\n", "import xarray as xr\n", "from sklearn.model_selection import train_test_split\n", "from sklearn.metrics import mean_absolute_error as MAE\n", "import matplotlib.pyplot as plt\n", "from tensorflow.keras import layers\n", "from tensorflow import keras\n", "import keras_tuner as kt\n", "r = np.random\n", "r.seed(42)" ] }, { "cell_type": "code", "execution_count": 34, "id": "fed09a5d", "metadata": {}, "outputs": [ { "data": { "text/html": [ "
<xarray.Dataset> Size: 566MB\n",
"Dimensions: (sample: 189326, x: 27, y: 27)\n",
"Coordinates:\n",
" * sample (sample) int32 757kB 0 1 2 3 4 ... 189322 189323 189324 189325\n",
" * x (x) int32 108B 0 1 2 3 4 5 6 7 8 ... 18 19 20 21 22 23 24 25 26\n",
" * y (y) int32 108B 0 1 2 3 4 5 6 7 8 ... 18 19 20 21 22 23 24 25 26\n",
"Data variables:\n",
" images (sample, x, y) float32 552MB ...\n",
" labels (sample) float64 2MB ...\n",
" vx (sample) float64 2MB ...\n",
" vy (sample) float64 2MB ...\n",
" v (sample) float64 2MB ...\n",
" smb (sample) float64 2MB ...\n",
" z (sample) float64 2MB ...\n",
" s (sample) float64 2MB ...\n",
" temp (sample) float64 2MB ...\n",
" gridCellId (sample) int32 757kB ...\n",
"Attributes:\n",
" description: CNN data with elevation images. N = 189326. Scalar features...