diff --git a/example.ipynb b/example.ipynb
index f54f8edd..163b0234 100644
--- a/example.ipynb
+++ b/example.ipynb
@@ -155,377 +155,6 @@
     "train_args = ['', epochs, log_step]"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": 32,
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/html": [
-       "<div>\n",
-       "<style scoped>\n",
-       "    .dataframe tbody tr th:only-of-type {\n",
-       "        vertical-align: middle;\n",
-       "    }\n",
-       "\n",
-       "    .dataframe tbody tr th {\n",
-       "        vertical-align: top;\n",
-       "    }\n",
-       "\n",
-       "    .dataframe thead th {\n",
-       "        text-align: right;\n",
-       "    }\n",
-       "</style>\n",
-       "<table border=\"1\" class=\"dataframe\">\n",
-       "  <thead>\n",
-       "    <tr style=\"text-align: right;\">\n",
-       "      <th></th>\n",
-       "      <th>V14</th>\n",
-       "      <th>V4</th>\n",
-       "      <th>V10</th>\n",
-       "      <th>V17</th>\n",
-       "      <th>Time</th>\n",
-       "      <th>V12</th>\n",
-       "      <th>V26</th>\n",
-       "      <th>Amount</th>\n",
-       "      <th>V21</th>\n",
-       "      <th>V8</th>\n",
-       "      <th>...</th>\n",
-       "      <th>V25</th>\n",
-       "      <th>V24</th>\n",
-       "      <th>V18</th>\n",
-       "      <th>V2</th>\n",
-       "      <th>V1</th>\n",
-       "      <th>V5</th>\n",
-       "      <th>V15</th>\n",
-       "      <th>V9</th>\n",
-       "      <th>V23</th>\n",
-       "      <th>Class_1</th>\n",
-       "    </tr>\n",
-       "  </thead>\n",
-       "  <tbody>\n",
-       "    <tr>\n",
-       "      <th>0</th>\n",
-       "      <td>-0.364140</td>\n",
-       "      <td>0.247360</td>\n",
-       "      <td>-0.284719</td>\n",
-       "      <td>-0.306976</td>\n",
-       "      <td>-0.208503</td>\n",
-       "      <td>-0.227107</td>\n",
-       "      <td>0.045838</td>\n",
-       "      <td>-0.199245</td>\n",
-       "      <td>0.071872</td>\n",
-       "      <td>0.148524</td>\n",
-       "      <td>...</td>\n",
-       "      <td>0.004413</td>\n",
-       "      <td>0.047312</td>\n",
-       "      <td>-0.004068</td>\n",
-       "      <td>0.136173</td>\n",
-       "      <td>-0.127201</td>\n",
-       "      <td>-0.038863</td>\n",
-       "      <td>0.038433</td>\n",
-       "      <td>-0.282337</td>\n",
-       "      <td>-0.075195</td>\n",
-       "      <td>0</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>1</th>\n",
-       "      <td>-0.166205</td>\n",
-       "      <td>0.152887</td>\n",
-       "      <td>-0.077801</td>\n",
-       "      <td>0.071059</td>\n",
-       "      <td>-0.208294</td>\n",
-       "      <td>-0.067495</td>\n",
-       "      <td>-0.022004</td>\n",
-       "      <td>0.181204</td>\n",
-       "      <td>0.090903</td>\n",
-       "      <td>-0.014051</td>\n",
-       "      <td>...</td>\n",
-       "      <td>0.051311</td>\n",
-       "      <td>-0.057084</td>\n",
-       "      <td>0.212486</td>\n",
-       "      <td>-0.181909</td>\n",
-       "      <td>-0.150359</td>\n",
-       "      <td>0.099108</td>\n",
-       "      <td>0.240807</td>\n",
-       "      <td>-0.019959</td>\n",
-       "      <td>0.230717</td>\n",
-       "      <td>0</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>2</th>\n",
-       "      <td>-0.147407</td>\n",
-       "      <td>0.155290</td>\n",
-       "      <td>-0.148970</td>\n",
-       "      <td>-0.495555</td>\n",
-       "      <td>-0.195564</td>\n",
-       "      <td>-0.387155</td>\n",
-       "      <td>-0.119771</td>\n",
-       "      <td>0.137843</td>\n",
-       "      <td>-0.039545</td>\n",
-       "      <td>-0.044662</td>\n",
-       "      <td>...</td>\n",
-       "      <td>-0.033824</td>\n",
-       "      <td>-0.024992</td>\n",
-       "      <td>-0.298661</td>\n",
-       "      <td>0.120709</td>\n",
-       "      <td>-0.126902</td>\n",
-       "      <td>-0.060265</td>\n",
-       "      <td>-0.078791</td>\n",
-       "      <td>-0.016853</td>\n",
-       "      <td>0.026953</td>\n",
-       "      <td>0</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>3</th>\n",
-       "      <td>-0.530328</td>\n",
-       "      <td>0.175246</td>\n",
-       "      <td>-0.518688</td>\n",
-       "      <td>-1.186117</td>\n",
-       "      <td>-0.187380</td>\n",
-       "      <td>-0.529034</td>\n",
-       "      <td>-0.150364</td>\n",
-       "      <td>0.059165</td>\n",
-       "      <td>0.079316</td>\n",
-       "      <td>-0.031046</td>\n",
-       "      <td>...</td>\n",
-       "      <td>0.045752</td>\n",
-       "      <td>-0.019474</td>\n",
-       "      <td>-0.569714</td>\n",
-       "      <td>0.089320</td>\n",
-       "      <td>-0.187225</td>\n",
-       "      <td>-0.082049</td>\n",
-       "      <td>-0.006559</td>\n",
-       "      <td>-0.017755</td>\n",
-       "      <td>-0.070646</td>\n",
-       "      <td>0</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>4</th>\n",
-       "      <td>-0.485576</td>\n",
-       "      <td>0.285807</td>\n",
-       "      <td>-0.248664</td>\n",
-       "      <td>0.978364</td>\n",
-       "      <td>-0.185638</td>\n",
-       "      <td>-0.308949</td>\n",
-       "      <td>0.115880</td>\n",
-       "      <td>-0.153064</td>\n",
-       "      <td>-0.051722</td>\n",
-       "      <td>-0.053250</td>\n",
-       "      <td>...</td>\n",
-       "      <td>0.314021</td>\n",
-       "      <td>-0.228582</td>\n",
-       "      <td>0.383094</td>\n",
-       "      <td>0.225762</td>\n",
-       "      <td>0.066542</td>\n",
-       "      <td>0.270500</td>\n",
-       "      <td>-0.042050</td>\n",
-       "      <td>-0.121053</td>\n",
-       "      <td>-0.105072</td>\n",
-       "      <td>0</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>...</th>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "      <td>...</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>487</th>\n",
-       "      <td>-0.523543</td>\n",
-       "      <td>0.121158</td>\n",
-       "      <td>-0.612683</td>\n",
-       "      <td>-0.475608</td>\n",
-       "      <td>0.164417</td>\n",
-       "      <td>-0.342881</td>\n",
-       "      <td>0.152349</td>\n",
-       "      <td>0.164564</td>\n",
-       "      <td>0.106174</td>\n",
-       "      <td>0.066094</td>\n",
-       "      <td>...</td>\n",
-       "      <td>0.104685</td>\n",
-       "      <td>-0.057246</td>\n",
-       "      <td>-0.154118</td>\n",
-       "      <td>0.071608</td>\n",
-       "      <td>-0.113806</td>\n",
-       "      <td>-0.113019</td>\n",
-       "      <td>0.039830</td>\n",
-       "      <td>-0.204329</td>\n",
-       "      <td>0.104693</td>\n",
-       "      <td>0</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>488</th>\n",
-       "      <td>-0.427730</td>\n",
-       "      <td>0.100689</td>\n",
-       "      <td>-0.336493</td>\n",
-       "      <td>-0.350142</td>\n",
-       "      <td>0.165773</td>\n",
-       "      <td>-0.237416</td>\n",
-       "      <td>0.144464</td>\n",
-       "      <td>-0.161527</td>\n",
-       "      <td>0.052358</td>\n",
-       "      <td>0.017518</td>\n",
-       "      <td>...</td>\n",
-       "      <td>0.101398</td>\n",
-       "      <td>-0.023974</td>\n",
-       "      <td>-0.082700</td>\n",
-       "      <td>0.084029</td>\n",
-       "      <td>0.079401</td>\n",
-       "      <td>0.031137</td>\n",
-       "      <td>-0.070237</td>\n",
-       "      <td>-0.104976</td>\n",
-       "      <td>-0.024638</td>\n",
-       "      <td>0</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>489</th>\n",
-       "      <td>-0.347682</td>\n",
-       "      <td>0.040168</td>\n",
-       "      <td>-0.362882</td>\n",
-       "      <td>-0.519358</td>\n",
-       "      <td>0.165799</td>\n",
-       "      <td>-0.220426</td>\n",
-       "      <td>0.099463</td>\n",
-       "      <td>0.074928</td>\n",
-       "      <td>0.102687</td>\n",
-       "      <td>0.126219</td>\n",
-       "      <td>...</td>\n",
-       "      <td>-0.138295</td>\n",
-       "      <td>-0.005116</td>\n",
-       "      <td>-0.061707</td>\n",
-       "      <td>0.071662</td>\n",
-       "      <td>-0.061751</td>\n",
-       "      <td>-0.081511</td>\n",
-       "      <td>-0.080171</td>\n",
-       "      <td>-0.056997</td>\n",
-       "      <td>0.029946</td>\n",
-       "      <td>0</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>490</th>\n",
-       "      <td>-0.507529</td>\n",
-       "      <td>0.125210</td>\n",
-       "      <td>-0.571613</td>\n",
-       "      <td>-0.479788</td>\n",
-       "      <td>0.169878</td>\n",
-       "      <td>-0.326933</td>\n",
-       "      <td>0.122500</td>\n",
-       "      <td>0.138998</td>\n",
-       "      <td>0.080595</td>\n",
-       "      <td>0.108006</td>\n",
-       "      <td>...</td>\n",
-       "      <td>-0.065488</td>\n",
-       "      <td>-0.040280</td>\n",
-       "      <td>-0.172473</td>\n",
-       "      <td>0.032122</td>\n",
-       "      <td>-0.152446</td>\n",
-       "      <td>-0.061618</td>\n",
-       "      <td>0.009960</td>\n",
-       "      <td>-0.157797</td>\n",
-       "      <td>-0.073768</td>\n",
-       "      <td>0</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>491</th>\n",
-       "      <td>-0.187473</td>\n",
-       "      <td>0.036138</td>\n",
-       "      <td>-0.082890</td>\n",
-       "      <td>0.110715</td>\n",
-       "      <td>0.172419</td>\n",
-       "      <td>0.077097</td>\n",
-       "      <td>-0.055913</td>\n",
-       "      <td>0.040476</td>\n",
-       "      <td>-0.021099</td>\n",
-       "      <td>-0.014108</td>\n",
-       "      <td>...</td>\n",
-       "      <td>0.058136</td>\n",
-       "      <td>-0.079923</td>\n",
-       "      <td>0.145454</td>\n",
-       "      <td>0.002641</td>\n",
-       "      <td>0.138765</td>\n",
-       "      <td>0.083551</td>\n",
-       "      <td>-0.092092</td>\n",
-       "      <td>0.056602</td>\n",
-       "      <td>-0.012857</td>\n",
-       "      <td>0</td>\n",
-       "    </tr>\n",
-       "  </tbody>\n",
-       "</table>\n",
-       "<p>492 rows × 31 columns</p>\n",
-       "</div>"
-      ],
-      "text/plain": [
-       "          V14        V4       V10       V17      Time       V12       V26  \\\n",
-       "0   -0.364140  0.247360 -0.284719 -0.306976 -0.208503 -0.227107  0.045838   \n",
-       "1   -0.166205  0.152887 -0.077801  0.071059 -0.208294 -0.067495 -0.022004   \n",
-       "2   -0.147407  0.155290 -0.148970 -0.495555 -0.195564 -0.387155 -0.119771   \n",
-       "3   -0.530328  0.175246 -0.518688 -1.186117 -0.187380 -0.529034 -0.150364   \n",
-       "4   -0.485576  0.285807 -0.248664  0.978364 -0.185638 -0.308949  0.115880   \n",
-       "..        ...       ...       ...       ...       ...       ...       ...   \n",
-       "487 -0.523543  0.121158 -0.612683 -0.475608  0.164417 -0.342881  0.152349   \n",
-       "488 -0.427730  0.100689 -0.336493 -0.350142  0.165773 -0.237416  0.144464   \n",
-       "489 -0.347682  0.040168 -0.362882 -0.519358  0.165799 -0.220426  0.099463   \n",
-       "490 -0.507529  0.125210 -0.571613 -0.479788  0.169878 -0.326933  0.122500   \n",
-       "491 -0.187473  0.036138 -0.082890  0.110715  0.172419  0.077097 -0.055913   \n",
-       "\n",
-       "       Amount       V21        V8  ...       V25       V24       V18  \\\n",
-       "0   -0.199245  0.071872  0.148524  ...  0.004413  0.047312 -0.004068   \n",
-       "1    0.181204  0.090903 -0.014051  ...  0.051311 -0.057084  0.212486   \n",
-       "2    0.137843 -0.039545 -0.044662  ... -0.033824 -0.024992 -0.298661   \n",
-       "3    0.059165  0.079316 -0.031046  ...  0.045752 -0.019474 -0.569714   \n",
-       "4   -0.153064 -0.051722 -0.053250  ...  0.314021 -0.228582  0.383094   \n",
-       "..        ...       ...       ...  ...       ...       ...       ...   \n",
-       "487  0.164564  0.106174  0.066094  ...  0.104685 -0.057246 -0.154118   \n",
-       "488 -0.161527  0.052358  0.017518  ...  0.101398 -0.023974 -0.082700   \n",
-       "489  0.074928  0.102687  0.126219  ... -0.138295 -0.005116 -0.061707   \n",
-       "490  0.138998  0.080595  0.108006  ... -0.065488 -0.040280 -0.172473   \n",
-       "491  0.040476 -0.021099 -0.014108  ...  0.058136 -0.079923  0.145454   \n",
-       "\n",
-       "           V2        V1        V5       V15        V9       V23  Class_1  \n",
-       "0    0.136173 -0.127201 -0.038863  0.038433 -0.282337 -0.075195        0  \n",
-       "1   -0.181909 -0.150359  0.099108  0.240807 -0.019959  0.230717        0  \n",
-       "2    0.120709 -0.126902 -0.060265 -0.078791 -0.016853  0.026953        0  \n",
-       "3    0.089320 -0.187225 -0.082049 -0.006559 -0.017755 -0.070646        0  \n",
-       "4    0.225762  0.066542  0.270500 -0.042050 -0.121053 -0.105072        0  \n",
-       "..        ...       ...       ...       ...       ...       ...      ...  \n",
-       "487  0.071608 -0.113806 -0.113019  0.039830 -0.204329  0.104693        0  \n",
-       "488  0.084029  0.079401  0.031137 -0.070237 -0.104976 -0.024638        0  \n",
-       "489  0.071662 -0.061751 -0.081511 -0.080171 -0.056997  0.029946        0  \n",
-       "490  0.032122 -0.152446 -0.061618  0.009960 -0.157797 -0.073768        0  \n",
-       "491  0.002641  0.138765  0.083551 -0.092092  0.056602 -0.012857        0  \n",
-       "\n",
-       "[492 rows x 31 columns]"
-      ]
-     },
-     "execution_count": 32,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "label"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": 33,