self.leakage_model = model_trained