miniworld.utils.DataClass.MiniWorldLabelClass¶
- class miniworld.utils.DataClass.MiniWorldLabelClass(label_dict=None)[source]¶
- __init__(label_dict=None)[source]¶
- label_dict :
- {
‘sequence’ : torch.tensor # (L_chain, NUM_CLASSES) ‘structure’ : {
‘xyz’ : torch.tensor, # (L_chain, 27, 3) ‘atom_mask’ : torch.tensor, # (L_chain, 27) ‘position_mask’ : torch.tensor, # (L_chain, 27) ‘has_multiple_chains’ : bool, ‘has_multiple_models’ : bool
}, ‘occupancy’ : occupancy,
}
Methods
__init__([label_dict])label_dict :