From 4f3b16e16319aba54d7f67364d228c96d23fa460 Mon Sep 17 00:00:00 2001 From: danielenricocahall Date: Wed, 13 Jan 2021 19:08:32 -0500 Subject: [PATCH] remove get_train_config --- elephas/spark_model.py | 7 ------- 1 file changed, 7 deletions(-) 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,