baby.training.smoothing_model_trainer.SmoothingModelTrainer¶
- class baby.training.smoothing_model_trainer.SmoothingModelTrainer(save_dir, stats_file, model_file)¶
Bases:
object- Attributes
- model
- stats
Methods
fit
generate_smoothing_sigma_stats
plot_fitted_model
- __init__(save_dir, stats_file, model_file)¶
Methods
__init__(save_dir, stats_file, model_file)fit([filt])generate_smoothing_sigma_stats(train_gen, ...)plot_fitted_model()Attributes
modelstats