Update SVDKL gridinterp regression notebook
},
{
"cell_type": "code",
- "execution_count": null,
+ "execution_count": 5,
"metadata": {},
"outputs": [],
"source": [
},
{
"cell_type": "code",
- "execution_count": null,
+ "execution_count": 6,
"metadata": {},
"outputs": [],
"source": [
},
{
"cell_type": "code",
- "execution_count": null,
+ "execution_count": 7,
"metadata": {
"scrolled": true
},
"Epoch 3 [169/170] - Loss: 14.799 [-14.798, 0.001, -0.000]\n",
"Epoch 4 [169/170] - Loss: 11.800 [-11.799, 0.001, -0.000]\n",
"Epoch 5 [169/170] - Loss: 9.114 [-9.113, 0.001, -0.000]\n",
- "Epoch 6 [169/170] - Loss: 7.376 [-7.375, 0.001, -0.000]\n"
+ "Epoch 6 [169/170] - Loss: 7.376 [-7.375, 0.001, -0.000]\n",
+ "Epoch 7 [169/170] - Loss: 7.074 [-7.074, 0.001, -0.000]\n",
+ "Epoch 8 [169/170] - Loss: 6.721 [-6.720, 0.001, -0.000]\n"
]
}
],
},
{
"cell_type": "code",
- "execution_count": null,
+ "execution_count": 8,
"metadata": {},
"outputs": [],
"source": [
},
{
"cell_type": "code",
- "execution_count": null,
+ "execution_count": 9,
"metadata": {},
- "outputs": [],
+ "outputs": [
+ {
+ "name": "stdout",
+ "output_type": "stream",
+ "text": [
+ "Test MAE: 11.856119155883789\n"
+ ]
+ }
+ ],
"source": [
"print('Test MAE: {}'.format(torch.mean(torch.abs(preds.mean - test_y))))"
]
