fix targets in learn function

PrincetonUniversity · Feb 21, 2025 · b0a63e3 · b0a63e3
1 parent b48c2b6
commit b0a63e3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Scripts/Examples/Lena Rumelhart script.py b/Scripts/Examples/Lena Rumelhart script.py
@@ -209,9 +209,9 @@ def gen_input_vals(nouns, relations):
                                          rels_in: rels_onehot[rel_out],
                                          },
                                  targets={out_sig_I: truth_nouns[noun],
-                                          out_sig_is: truth_is[noun],
-                                          out_sig_has: truth_has[noun],
-                                          out_sig_can: truth_can[noun]
+                                          out_sig_is: truth_is[noun] if rel_out == 0 else np.zeros(len(is_list)),
+                                          out_sig_has: truth_has[noun] if rel_out == 1 else np.zeros(len(has_list)),
+                                          out_sig_can: truth_can[noun] if rel_out == 2 else np.zeros(len(can_list)),
                                           },
                                  )