Addition of base module with abstract classes models.

muammar · Dec 25, 2019 · d5f2cae · d5f2cae
1 parent 5cb63c4
commit d5f2cae
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/ml4chem/models/base.py b/ml4chem/models/base.py
@@ -0,0 +1,19 @@
+from abc import ABC, abstractmethod
+
+
+class DeepLearningModel(ABC):
+    @abstractmethod
+    def name(cls):
+        pass
+
+    @abstractmethod
+    def __init__(self, **kwargs):
+        pass
+
+    @abstractmethod
+    def prepare_model(self, **kwargs):
+        pass
+
+    @abstractmethod
+    def forward(self, X):
+        pass