Revert to 64 channels embedding

25721808 · David Landry · 0118c5bb · 25721808 · 25721808 · 25721808
Commit 25721808 authored 3 years ago by David Landry
--- a/notebooks/yearly_eval.ipynb
+++ b/notebooks/yearly_eval.ipynb
@@ -56,83 +56,83 @@
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>2000</th>\n",
-       "      <td>0.064819</td>\n",
+       "      <td>0.063627</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2001</th>\n",
-       "      <td>0.037976</td>\n",
+       "      <td>0.036329</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2002</th>\n",
-       "      <td>0.032928</td>\n",
+       "      <td>0.031632</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2003</th>\n",
-       "      <td>0.014985</td>\n",
+       "      <td>0.014625</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2004</th>\n",
-       "      <td>0.026293</td>\n",
+       "      <td>0.025528</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2005</th>\n",
-       "      <td>0.032006</td>\n",
+       "      <td>0.031132</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2006</th>\n",
-       "      <td>0.027906</td>\n",
+       "      <td>0.027506</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2007</th>\n",
-       "      <td>0.027021</td>\n",
+       "      <td>0.028149</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2008</th>\n",
-       "      <td>0.033429</td>\n",
+       "      <td>0.035192</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2009</th>\n",
-       "      <td>0.018628</td>\n",
+       "      <td>0.019445</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2010</th>\n",
-       "      <td>0.040712</td>\n",
+       "      <td>0.040541</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2011</th>\n",
-       "      <td>0.024528</td>\n",
+       "      <td>0.026224</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2012</th>\n",
-       "      <td>0.026787</td>\n",
+       "      <td>0.029145</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2013</th>\n",
-       "      <td>0.023060</td>\n",
+       "      <td>0.024344</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2014</th>\n",
-       "      <td>0.016251</td>\n",
+       "      <td>0.016472</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2015</th>\n",
-       "      <td>0.032162</td>\n",
+       "      <td>0.032913</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2016</th>\n",
-       "      <td>0.039090</td>\n",
+       "      <td>0.041742</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2017</th>\n",
-       "      <td>0.014229</td>\n",
+       "      <td>0.014178</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2018</th>\n",
-       "      <td>0.015166</td>\n",
+       "      <td>0.018790</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2019</th>\n",
-       "      <td>0.025263</td>\n",
+       "      <td>0.025822</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
@@ -141,26 +141,26 @@
      "text/plain": [
       "          RPSS\n",
       "year          \n",
-       "2000  0.064819\n",
-       "2001  0.037976\n",
-       "2002  0.032928\n",
-       "2003  0.014985\n",
-       "2004  0.026293\n",
-       "2005  0.032006\n",
-       "2006  0.027906\n",
-       "2007  0.027021\n",
-       "2008  0.033429\n",
-       "2009  0.018628\n",
-       "2010  0.040712\n",
-       "2011  0.024528\n",
-       "2012  0.026787\n",
-       "2013  0.023060\n",
-       "2014  0.016251\n",
-       "2015  0.032162\n",
-       "2016  0.039090\n",
-       "2017  0.014229\n",
-       "2018  0.015166\n",
-       "2019  0.025263"
+       "2000  0.063627\n",
+       "2001  0.036329\n",
+       "2002  0.031632\n",
+       "2003  0.014625\n",
+       "2004  0.025528\n",
+       "2005  0.031132\n",
+       "2006  0.027506\n",
+       "2007  0.028149\n",
+       "2008  0.035192\n",
+       "2009  0.019445\n",
+       "2010  0.040541\n",
+       "2011  0.026224\n",
+       "2012  0.029145\n",
+       "2013  0.024344\n",
+       "2014  0.016472\n",
+       "2015  0.032913\n",
+       "2016  0.041742\n",
+       "2017  0.014178\n",
+       "2018  0.018790\n",
+       "2019  0.025822"
      ]
     },
     "execution_count": 3,
@@ -209,7 +209,7 @@
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>2020</th>\n",
-       "      <td>0.043799</td>\n",
+       "      <td>0.045947</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
@@ -218,7 +218,7 @@
      "text/plain": [
       "          RPSS\n",
       "year          \n",
-       "2020  0.043799"
+       "2020  0.045947"
      ]
     },
     "execution_count": 4,

 %% Cell type:code id: tags:

 ``` python
 import xarray as xr
 from scripts import skill_by_year
 ```

 %% Cell type:code id: tags:

 ``` python
 TRAIN_PREDS = '../submissions/ML_prediction_train.nc'
 TEST_PREDS = '../submissions/ML_prediction_2020.nc'
 ```

 %% Cell type:code id: tags:

 ``` python
 train_preds = xr.open_dataset(TRAIN_PREDS)
 skill_by_year(train_preds)
 ```

 %% Output

              RPSS
    year
-    2000  0.064819
-    2001  0.037976
-    2002  0.032928
-    2003  0.014985
-    2004  0.026293
-    2005  0.032006
-    2006  0.027906
-    2007  0.027021
-    2008  0.033429
-    2009  0.018628
-    2010  0.040712
-    2011  0.024528
-    2012  0.026787
-    2013  0.023060
-    2014  0.016251
-    2015  0.032162
-    2016  0.039090
-    2017  0.014229
-    2018  0.015166
-    2019  0.025263
+    2000  0.063627
+    2001  0.036329
+    2002  0.031632
+    2003  0.014625
+    2004  0.025528
+    2005  0.031132
+    2006  0.027506
+    2007  0.028149
+    2008  0.035192
+    2009  0.019445
+    2010  0.040541
+    2011  0.026224
+    2012  0.029145
+    2013  0.024344
+    2014  0.016472
+    2015  0.032913
+    2016  0.041742
+    2017  0.014178
+    2018  0.018790
+    2019  0.025822

 %% Cell type:code id: tags:

 ``` python
 test_preds = xr.open_dataset(TEST_PREDS)
 skill_by_year(test_preds)
 ```

 %% Output

              RPSS
    year
-    2020  0.043799
+    2020  0.045947

--- a/submissions/ML_prediction_2020.nc
+++ b/submissions/ML_prediction_2020.nc
--- a/submissions/ML_prediction_train.nc
+++ b/submissions/ML_prediction_train.nc