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