--added models modules in the init

manujosephv · manujosephv · commit 16ed313d80b2 · 2021-03-03T15:02:30.000+05:30
diff --git a/.vscode/settings.json b/.vscode/settings.json
@@ -2,5 +2,6 @@
     "python.pythonPath": "env\\Scripts\\python.exe",
     "python.testing.unittestEnabled": false,
     "python.testing.nosetestsEnabled": false,
-    "python.testing.pytestEnabled": true
+    "python.testing.pytestEnabled": false,
+    "restructuredtext.languageServer.disabled": true
 }
diff --git a/examples/regression_with_MDN.ipynb b/examples/regression_with_MDN.ipynb
diff --git a/pytorch_tabular/models/__init__.py b/pytorch_tabular/models/__init__.py
@@ -10,6 +10,7 @@
     NODEMDN,
 )
 from .base_model import BaseModel
+from . import category_embedding, node, mixture_density, tabnet
 
 __all__ = [
     "CategoryEmbeddingModel",
@@ -24,5 +25,9 @@
     "MixtureDensityHead",
     "MixtureDensityHeadConfig",
     "NODEMDNConfig",
-    "NODEMDN"
+    "NODEMDN",
+    "category_embedding",
+    "node",
+    "mixture_density",
+    "tabnet",
 ]

Original file line number	Diff line number	Diff line change
`@@ -2,5 +2,6 @@`
`2`	`2`	`"python.pythonPath": "env\\Scripts\\python.exe",`
`3`	`3`	`"python.testing.unittestEnabled": false,`
`4`	`4`	`"python.testing.nosetestsEnabled": false,`
`5`		`- "python.testing.pytestEnabled": true`
	`5`	`+ "python.testing.pytestEnabled": false,`
	`6`	`+ "restructuredtext.languageServer.disabled": true`
`6`	`7`	`}`