unimib-thesis-mad
/
PyCTBN
Archived
1
0
Fork 0
Code Releases Activity

Undefine node children time

Browse Source
master
Pietro 4 years ago
parent c71699d267
commit 710a392fac
1 changed files with 5 additions and 1 deletions
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Show Stats Download Patch File Download Diff File
  1. 6
      PyCTBN/PyCTBN/structure_graph/trajectory_generator.py

6
PyCTBN/PyCTBN/structure_graph/trajectory_generator.py
Unescape View File

@ -11,7 +11,6 @@ from numpy import random
class TrajectoryGenerator(object): class TrajectoryGenerator(object):
def __init__(self, importer: AbstractImporter): def __init__(self, importer: AbstractImporter):
self._importer = importer self._importer = importer
self._importer.import_data(0)
self._vnames = self._importer._df_variables.iloc[:, 0].to_list() self._vnames = self._importer._df_variables.iloc[:, 0].to_list()
@ -19,6 +18,8 @@ class TrajectoryGenerator(object):
for v in self._vnames: for v in self._vnames:
self._parents[v] = self._importer._df_structure.where(self._importer._df_structure["To"] == v).dropna()["From"].tolist() self._parents[v] = self._importer._df_structure.where(self._importer._df_structure["To"] == v).dropna()["From"].tolist()
print(self._parents)
self._cims = {} self._cims = {}
sampled_cims = self._importer._raw_data[0]["dyn.cims"] sampled_cims = self._importer._raw_data[0]["dyn.cims"]
for v in sampled_cims.keys(): for v in sampled_cims.keys():
@ -92,6 +93,9 @@ class TrajectoryGenerator(object):
# undefine variable time # undefine variable time
time[next] = np.NaN time[next] = np.NaN
for i, v in enumerate(self._parents):
if self._vnames[next] in self._parents[v]:
time[i] = np.NaN
def to_json(self): def to_json(self):
return json.loads(self._generated_trajectory.to_json(orient="records")) return json.loads(self._generated_trajectory.to_json(orient="records"))