Update TwoBranchCNN1D.py

TwoBranchCNN1D.py

@@ -42,6 +42,7 @@ class TwoBranchCNN1D:
         #self.output_directory = output_directory
         self.model = self.build_model(input_shape1, input_shape2, nb_classes)
         self.callbacks = None
+        return self.model
 
     def build_model(self, input_shape1, input_shape2, nb_classes):
         n_feature_maps = 128
@@ -72,7 +73,7 @@ class TwoBranchCNN1D:
         self.callbacks = [reduce_lr]#, model_checkpoint]
 
         return model#, encoder
-
+    '''
     def fit(self, x_train1, x_train2, y_train, batch_size, nb_epochs):
         mini_batch_size = int(min(x_train.shape[0] / 10, batch_size))
         start_time = time.time()
@@ -85,3 +86,4 @@ class TwoBranchCNN1D:
         start_time = time.time()
         y_pred = model.predict([x_test1, x_test2])
         return y_pred
+    '''
\ No newline at end of file