bsitep.utils package

Submodules

bsitep.utils.data_utils module

bsitep.utils.data_utils.prepare_data(input_file, file_format)

bsitep.utils.data_utils.process_protein_data(data, file_format)

bsitep.utils.model_utils module

bsitep.utils.model_utils.load_model(model_path)

bsitep.utils.pocket_utils module

bsitep.utils.pocket_utils.get_pockets_segmentation(density, threshold=0.5, min_size=50, scale=0.5)

bsitep.utils.pocket_utils.save_pocket_mol(density, origin, step, mol, dist_cutoff=4.5, expand_residue=True)

Module contents