diff --git a/TwoBranchCNN1D.py b/TwoBranchCNN1D.py index 97ea57c9d2e81b369c4d21e600305af5bbe88690..3e8a2a9bd230d830f5e9224e987191a36178a9bc 100644 --- a/TwoBranchCNN1D.py +++ b/TwoBranchCNN1D.py @@ -39,15 +39,15 @@ def getBranch(input_layer, n_filters, dropOut, suffix): -def build_model(self, input_shape1, input_shape2, nb_classes): +def build_model(input_shape1, input_shape2, nb_classes): n_feature_maps = 128 dropOut = 0.2 input_layer1 = tfk.layers.Input(name="input1", shape=input_shape1) input_layer2 = tfk.layers.Input(name="input2", shape=input_shape2) - features1 = self.getBranch(input_layer1, n_feature_maps, dropOut, "input1") - features2 = self.getBranch(input_layer2, n_feature_maps, dropOut, "input2") + features1 = getBranch(input_layer1, n_feature_maps, dropOut, "input1") + features2 = getBranch(input_layer2, n_feature_maps, dropOut, "input2") features = tfk.Concatenation(name="concat_layer")([features1, features2]) dense_layer = tfk.layers.Dense(512, activation='relu')(features)