1
0
Fork 0

Fixed edges generation problem in constraint

master
philipMartini 4 years ago committed by GitHub
parent 78cfd9b081
commit c59287fdc2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 2
      PyCTBN/PyCTBN/estimators/structure_constraint_based_estimator.py

@ -94,7 +94,7 @@ class StructureConstraintBasedEstimator(StructureEstimator):
l2 = list(self._nodes[bool_mask2]) l2 = list(self._nodes[bool_mask2])
indxs2 = self._nodes_indxs[bool_mask2] indxs2 = self._nodes_indxs[bool_mask2]
vals2 = self._nodes_vals[bool_mask2] vals2 = self._nodes_vals[bool_mask2]
eds2 = list(itertools.product(p_set, test_child)) eds2 = list(itertools.product(p_set, [test_child]))
s2 = Structure(l2, indxs2, vals2, eds2, tot_vars_count) s2 = Structure(l2, indxs2, vals2, eds2, tot_vars_count)
g2 = NetworkGraph(s2) g2 = NetworkGraph(s2)
g2.fast_init(test_child) g2.fast_init(test_child)