Merge pull request deepchem#3586 from shreyasvinaya/GAN

Gan
GreatRSingh · Oct 8, 2023 · 05a6608 · 05a6608
2 parents 039c6a7 + 4164305
commit 05a6608
Show file tree

Hide file tree

Showing 4 changed files with 634 additions and 0 deletions.
diff --git a/deepchem/models/torch_models/__init__.py b/deepchem/models/torch_models/__init__.py
@@ -9,6 +9,7 @@
 from deepchem.models.torch_models.cgcnn import CGCNN, CGCNNModel
 from deepchem.models.torch_models.gat import GAT, GATModel
 from deepchem.models.torch_models.gcn import GCN, GCNModel
+from deepchem.models.torch_models.gan import GAN
 from deepchem.models.torch_models.infograph import InfoGraphStar, InfoGraphStarModel, InfoGraphEncoder, GINEncoder, InfoGraph, InfoGraphModel, InfoGraphEncoder
 from deepchem.models.torch_models.mpnn import MPNN, MPNNModel
 from deepchem.models.torch_models.lcnn import LCNN, LCNNModel