swapping torch code for tutorial (deepchem#4200)

Dragonwagon18 · Dec 18, 2024 · 7207078 · 7207078
1 parent 216faf6
commit 7207078
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/tutorials/Introduction_to_Graph_Convolutions.ipynb b/examples/tutorials/Introduction_to_Graph_Convolutions.ipynb
@@ -121,7 +121,8 @@
    ],
    "source": [
     "n_tasks = len(tasks)\n",
-    "model = dc.models.GraphConvModel(n_tasks, mode='classification')\n",
+    "num_features = train_dataset.X[0].get_atom_features().shape[1]\n",
+    "model = dc.models.torch_models.GraphConvModel(n_tasks, mode='classification',number_input_features=[num_features,64])\n",
     "model.fit(train_dataset, nb_epoch=50)"
    ]
   },