diff --git a/elephas/spark_model.py b/elephas/spark_model.py index c99e6ec..b64e9dd 100644 --- a/elephas/spark_model.py +++ b/elephas/spark_model.py @@ -65,13 +65,6 @@ def __init__(self, model, mode='asynchronous', frequency='epoch', parameter_ser self.parameter_server = factory.create_server(self.serialized_model, self.port, mode=self.mode) self.client = factory.create_client(self.port) - @staticmethod - def get_train_config(epochs, batch_size, verbose, validation_split): - return {'epochs': epochs, - 'batch_size': batch_size, - 'verbose': verbose, - 'validation_split': validation_split} - def get_config(self): base_config = { 'parameter_server_mode': self.parameter_server_mode,