added ensemble with one shift

calico · Oct 17, 2023 · dc6802a · dc6802a
1 parent e84e220
commit dc6802a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/baskerville/seqnn.py b/src/baskerville/seqnn.py
@@ -219,12 +219,12 @@ def build_embed(self, conv_layer_i: int, batch_norm: bool = True):
 
     def build_ensemble(self, ensemble_rc: bool = False, ensemble_shifts=[0]):
         """Build ensemble of models computing on augmented input sequences."""
-        if ensemble_rc or len(ensemble_shifts) > 1:
+        if ensemble_rc or len(ensemble_shifts) > 1 or int(ensemble_shifts[0]) != 0:
             # sequence input
             sequence = tf.keras.Input(shape=(self.seq_length, 4), name="sequence")
             sequences = [sequence]
 
-            if len(ensemble_shifts) > 1:
+            if len(ensemble_shifts) > 1 or int(ensemble_shifts[0]) != 0:
                 # generate shifted sequences
                 sequences = layers.EnsembleShift(ensemble_shifts)(sequences)