Skip to content

Commit 38e2f01

Browse files
Fix CEM example high level
1 parent 879dec4 commit 38e2f01

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/utilization/2_model/1_concept_embedding_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,8 +39,8 @@ def main():
3939
input_size=latent_dims,
4040
annotations=annotations,
4141
source_exogenous=LazyConstructor(LinearZU, exogenous_size=12),
42-
encoder=LazyConstructor(LinearUC),
43-
predictor=LazyConstructor(MixCUC),
42+
encoder=LazyConstructor(LinearUC, n_exogenous_per_concept=1),
43+
predictor=LazyConstructor(MixCUC, cardinalities=[1, 1]),
4444
use_source_exogenous=True)
4545

4646
# Inference Initialization

0 commit comments

Comments
 (0)