mergernet.core.experiment.backup_model#
- mergernet.core.experiment.backup_model(model: Model, dataset: Any, save_history: bool = True, save_test_preds: bool = True, save_dataset_config: bool = True, save_model: bool = True)[source]#
- Parameters:
model (tf.keras.Model) – The keras model to backup
dataset (Dataset) – The dataset instance
save_history (bool) – True if train history must be saved in github
save_test_preds (bool) – True if the predictions in test dataset must be saved in github
save_dataset_config (bool) – True if dataset_config must be saved in github
save_model (bool) – True if tensorflow model must be saved in google drive