1
0
Fork 0
master
Pietro 4 years ago
parent ff1765c3a2
commit 16f9b2d7c3
  1. 1
      .gitignore
  2. 9
      PyCTBN/PyCTBN/structure_graph/trajectory_generator.py

1
.gitignore vendored

@ -10,3 +10,4 @@ __pycache__
test.py test.py
test_1.py test_1.py
test1.json test1.json
test2.json

@ -66,6 +66,10 @@ class TrajectoryGenerator(object):
t = time[next] t = time[next]
if (max_tr != -1 and n_tr == max_tr) or (t_end != -1 and t >= t_end): if (max_tr != -1 and n_tr == max_tr) or (t_end != -1 and t >= t_end):
last_row = pd.DataFrame(sigma[-1:].values, columns = sigma.columns)
last_row.loc[0].values[:] = -1
last_row.loc[0].at["Time"] = round(t, 4)
sigma = sigma.append(last_row, ignore_index = True)
self._generated_trajectory = sigma self._generated_trajectory = sigma
return sigma return sigma
else: else:
@ -78,11 +82,6 @@ class TrajectoryGenerator(object):
else: else:
cim = cim_obj[0].cim cim = cim_obj[0].cim
""" print(self._vnames[next])
print([current_values.at[p] for p in self._parents[self._vnames[next]]])
print(current_values)
print(cim)
print() """
cim_row = np.array(cim[current_values.at[self._vnames[next]]]) cim_row = np.array(cim[current_values.at[self._vnames[next]]])
cim_row[current_values.at[self._vnames[next]]] = 0 cim_row[current_values.at[self._vnames[next]]] = 0
cim_row /= sum(cim_row) cim_row /= sum(cim_row)