def METHOD_NAME(self, parameters, config):
    self.set_parameters(parameters)
    [accuracy, loss], num_examples = train(model, train_data, epoch=2)
    results = {"accuracy": float(accuracy[1]), "loss": float(loss[1])}
    return self.get_parameters(config={}), num_examples, results