concept embedding model

Author: gdefe

conceptarium/conf/model/cem.yaml

@@ -0,0 +1,14 @@
+defaults:
+  - _commons
+  - _self_
+  
+# wrapper for 'joint' training mode
+_target_: "torch_concepts.nn.ConceptEmbeddingModel"
+
+task_names: ${dataset.default_task_names}
+
+exogenous_size: 16
+
+inference: 
+  _target_: "torch_concepts.nn.DeterministicInference"
+  _partial_: true

conceptarium/conf/model/cem_joint.yaml

@@ -0,0 +1,13 @@
+defaults:
+  - _commons
+  - _self_
+  
+_target_: "torch_concepts.nn.ConceptEmbeddingModel_Joint"
+
+task_names: ${dataset.default_task_names}
+
+exogenous_size: 16
+
+inference: 
+  _target_: "torch_concepts.nn.DeterministicInference"
+  _partial_: true