apply_train_type#
- radiosim.jets.jet.apply_train_type(train_type, jet_img, jet_comp, source_list)[source]#
Creating the y-data dependent on the training type.
- Parameters:
- train_type: str
‘list’: returns the components attributes only ‘gauss’: returns all components, background and list ‘clean’: returns sum of components and background (usage for softmax)
- jet_comps: ndarray
simulated jet components as an image
- source_list:
attributes of jet components
- Returns:
- y: ndarray
output data