XGBClassifier(base_score=None, booster=None, callbacks=None,\n",
" colsample_bylevel=None, colsample_bynode=None,\n",
" colsample_bytree=0.8, device=None, early_stopping_rounds=None,\n",
" enable_categorical=False, eval_metric=None, feature_types=None,\n",
@@ -1699,7 +1721,7 @@
"name": "stdout",
"output_type": "stream",
"text": [
- "gini_score for test dataset: 0.060025811573610754\n"
+ "gini_score for test dataset: 0.06491440835995244\n"
]
}
],
@@ -1783,8 +1805,8 @@
"name": "stdout",
"output_type": "stream",
"text": [
- "train data shape = (1144339, 977)\n",
- "test data shape = (382320, 977)\n"
+ "train data shape = (1145346, 980)\n",
+ "test data shape = (381313, 980)\n"
]
}
],
@@ -1882,8 +1904,8 @@
"y_test_tensor = torch.from_numpy(y_test_array)\n",
"val_dataset = TensorDataset(X_test_tensor, y_test_tensor)\n",
"\n",
- "train_loader = DataLoader(train_dataset, batch_size=32, shuffle=False)\n",
- "val_loader = DataLoader(val_dataset, batch_size=32, shuffle=False)"
+ "train_loader = DataLoader(train_dataset, batch_size=32, num_workers=13, shuffle=False)\n",
+ "val_loader = DataLoader(val_dataset, batch_size=32, num_workers=13, shuffle=False)"
]
},
{
@@ -1913,7 +1935,7 @@
"text": [
"NeuralNetClassifier(\n",
" (model): Sequential(\n",
- " (0): Linear(in_features=977, out_features=8, bias=True)\n",
+ " (0): Linear(in_features=980, out_features=8, bias=True)\n",
" (1): ReLU()\n",
" (2): Linear(in_features=8, out_features=1, bias=True)\n",
" )\n",
@@ -1926,7 +1948,7 @@
"name": "stderr",
"output_type": "stream",
"text": [
- "/Users/claypot/miniconda3/envs/ibisml-dev/lib/python3.12/site-packages/pytorch_lightning/trainer/connectors/logger_connector/logger_connector.py:75: Starting from v1.9.0, `tensorboardX` has been removed as a dependency of the `pytorch_lightning` package, due to potential conflicts with other packages in the ML ecosystem. For this reason, `logger=True` will use `CSVLogger` as the default logger, unless the `tensorboard` or `tensorboardX` packages are found. Please `pip install lightning[extra]` or one of them to enable TensorBoard support by default\n"
+ "/Users/jiting/anaconda3/envs/ibis-ml-dev/lib/python3.12/site-packages/pytorch_lightning/trainer/connectors/logger_connector/logger_connector.py:75: Starting from v1.9.0, `tensorboardX` has been removed as a dependency of the `pytorch_lightning` package, due to potential conflicts with other packages in the ML ecosystem. For this reason, `logger=True` will use `CSVLogger` as the default logger, unless the `tensorboard` or `tensorboardX` packages are found. Please `pip install lightning[extra]` or one of them to enable TensorBoard support by default\n"
]
}
],
@@ -1955,20 +1977,22 @@
"\n",
" | Name | Type | Params | Mode \n",
"------------------------------------------------------\n",
- "0 | model | Sequential | 7.8 K | train\n",
+ "0 | model | Sequential | 7.9 K | train\n",
"1 | loss | BCEWithLogitsLoss | 0 | train\n",
"2 | sigmoid | Sigmoid | 0 | train\n",
"------------------------------------------------------\n",
- "7.8 K Trainable params\n",
+ "7.9 K Trainable params\n",
"0 Non-trainable params\n",
- "7.8 K Total params\n",
- "0.031 Total estimated model params size (MB)\n"
+ "7.9 K Total params\n",
+ "0.031 Total estimated model params size (MB)\n",
+ "6 Modules in train mode\n",
+ "0 Modules in eval mode\n"
]
},
{
"data": {
"application/vnd.jupyter.widget-view+json": {
- "model_id": "9afcc7876b13405c9ef4d6b92b686523",
+ "model_id": "2cf7098fcd4c41f286c6059b3b170828",
"version_major": 2,
"version_minor": 0
},
@@ -1983,14 +2007,14 @@
"name": "stderr",
"output_type": "stream",
"text": [
- "/Users/claypot/miniconda3/envs/ibisml-dev/lib/python3.12/site-packages/pytorch_lightning/trainer/connectors/data_connector.py:424: The 'val_dataloader' does not have many workers which may be a bottleneck. Consider increasing the value of the `num_workers` argument` to `num_workers=7` in the `DataLoader` to improve performance.\n",
- "/Users/claypot/miniconda3/envs/ibisml-dev/lib/python3.12/site-packages/pytorch_lightning/trainer/connectors/data_connector.py:424: The 'train_dataloader' does not have many workers which may be a bottleneck. Consider increasing the value of the `num_workers` argument` to `num_workers=7` in the `DataLoader` to improve performance.\n"
+ "/Users/jiting/anaconda3/envs/ibis-ml-dev/lib/python3.12/site-packages/pytorch_lightning/trainer/connectors/data_connector.py:424: The 'val_dataloader' does not have many workers which may be a bottleneck. Consider increasing the value of the `num_workers` argument` to `num_workers=13` in the `DataLoader` to improve performance.\n",
+ "/Users/jiting/anaconda3/envs/ibis-ml-dev/lib/python3.12/site-packages/pytorch_lightning/trainer/connectors/data_connector.py:424: The 'train_dataloader' does not have many workers which may be a bottleneck. Consider increasing the value of the `num_workers` argument` to `num_workers=13` in the `DataLoader` to improve performance.\n"
]
},
{
"data": {
"application/vnd.jupyter.widget-view+json": {
- "model_id": "9ce4448968c24169bf92126a6e5893de",
+ "model_id": "e77639d880a543a9a35d04c4306faf89",
"version_major": 2,
"version_minor": 0
},
@@ -2000,41 +2024,6 @@
},
"metadata": {},
"output_type": "display_data"
- },
- {
- "data": {
- "application/vnd.jupyter.widget-view+json": {
- "model_id": "fa08058338ba401e9bc6c7a7243239a4",
- "version_major": 2,
- "version_minor": 0
- },
- "text/plain": [
- "Validation: | | 0/? [00:00, ?it/s]"
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- },
- {
- "data": {
- "application/vnd.jupyter.widget-view+json": {
- "model_id": "e4c0ad19f3354001889e9432dd2c83a8",
- "version_major": 2,
- "version_minor": 0
- },
- "text/plain": [
- "Validation: | | 0/? [00:00, ?it/s]"
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- },
- {
- "name": "stderr",
- "output_type": "stream",
- "text": [
- "`Trainer.fit` stopped: `max_epochs=2` reached.\n"
- ]
}
],
"source": [
@@ -2121,7 +2110,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
- "version": "3.12.3"
+ "version": "3.12.4"
}
},
"nbformat": 4,