remove get_train_config

maxpumperla · Jan 14, 2021 · 4f3b16e · 4f3b16e
1 parent 8e30175
commit 4f3b16e
Showing 1 changed file with 0 additions and 7 deletions.
diff --git 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,