def main(k, d, zmatrix_fn, stride, project):
hierarchical = Hierarchical.load_from_disk(zmatrix_fn)
assignments = hierarchical.get_assignments(k=k, cutoff_distance=d)
new_assignments = np.ones((project.n_trajs, project.traj_lengths.max()), dtype=np.int) * -1
new_assignments[:, ::stride] = assignments
return new_assignments
def main(k, d, zmatrix_fn, stride, project):
hierarchical = Hierarchical.load_from_disk(zmatrix_fn)
assignments = hierarchical.get_assignments(k=k, cutoff_distance=d)
new_assignments = np.ones(
(project.n_trajs, project.traj_lengths.max()), dtype=np.int) * -1
new_assignments[:, ::stride] = assignments
return new_assignments
def main(k, d, zmatrix_fn):
hierarchical = Hierarchical.load_from_disk(zmatrix_fn)
assignments = hierarchical.get_assignments(k=k, cutoff_distance=d)
return assignments
