ncem.estimators.EstimatorCVAE.train_aggressive

EstimatorCVAE.train_aggressive(aggressive_enc_patience: int = 10, aggressive_epochs: int = 5)

Train model aggressive.

Parameters

  • aggressive_enc_patience (int) – Number of epochs with no improvement in aggressive training. If unspecified, it will default to 10.

  • aggressive_epochs (int) – Integer number of times to iterate over the training data arrays in aggressive training. If unspecified, it will default to 5.