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