Merge pull request #17 from preinaj/save-model

Add support for save model offline

ADLStream/adlstream.py

@@ -116,6 +116,10 @@ def set_weights(self, w):
     def get_weights(self):
         return self.weights
 
+    def get_shape(self):
+        X = self.x_train
+        return np.asarray(X).shape
+
     def add(self, x, y=None):
         with self.data_lock:
             if x is not None:
@@ -253,6 +257,9 @@ def __init__(
         self.train_gpu_index = train_gpu_index
         self.predict_gpu_index = predict_gpu_index
         self.log_file = log_file
+        self.x_shape = None
+        self.output_size = None
+        self.weights = None
 
         self.manager = ADLStreamManager()
 
@@ -437,4 +444,22 @@ def run(self):
         process_predict.join()
         process_evaluator.join()
 
+        self.x_shape = context.get_shape()
+        self.output_size = context.get_output_size()
+        self.weights = context.get_weights()
+
         self.manager.shutdown()
+
+    def get_model(self):
+        from ADLStream.models import create_model
+
+        model = create_model(
+            self.model_architecture,
+            self.x_shape,
+            self.output_size,
+            self.model_loss,
+            self.model_optimizer,
+            **self.model_parameters
+        )
+        model.set_weights(self.weights)
+        return model