save_output#
- protopipe.mva.save_output(models, cam_id, factory, best_model, model_types, method_name, outdir)[source]#
Save model and data used to produce it per camera-type.
- Parameters
- models: dict
Dictionary of models with camera names as keys.
- cam_id: str
Name of the analyzed camera.
- factory: protopipe.mva.TrainModel
Wrapper around trained model containing references to train/test samples.
- best_model:
Fit of the model from factory.
- model_types: dict
Dictionary that maps type of model to method name.
- method_name: str
Name of the scikit-learn model.
- outdir: str
Path to output directory where to save the trained model and train/test samples.