miniworld.utils.output_visualize

Functions

TMalign(candidate, answer)

align_two_protein(protein1, protein2)

atom_to_pdb_line(atom, atom_xyz, ...)

output_to_pdb(output_dict[, save_dir])

protein_to_pdb(ID, protein[, crop_idx, save_dir])

tensor_to_pdb(ID, xyz, sequence, atom_mask)

visualize_2D_heatmap(heatmap[, file_name, ...])

visualize_2D_heatmaps(heatmap1, heatmap2[, ...])

visualize_2D_tensor(tensor[, saving_path])

visualize_t_map(t_map_list, l_idx[, ...])

t_map : list of tensor (L,L,n_atoms,3) (size = 2)