From 5714cd9d0b69ca393ea064b0522bda0bea4737c9 Mon Sep 17 00:00:00 2001 From: asistradition Date: Tue, 17 Oct 2023 14:59:36 -0400 Subject: [PATCH] Fix mask layering --- supirfactor_dynamical/models/chromatin_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/supirfactor_dynamical/models/chromatin_model.py b/supirfactor_dynamical/models/chromatin_model.py index 52ccf02..4dace78 100644 --- a/supirfactor_dynamical/models/chromatin_model.py +++ b/supirfactor_dynamical/models/chromatin_model.py @@ -54,7 +54,7 @@ def __init__( self.mask_input_weights( gene_peak_mask, - module=self.peak_encoder[1], + module=self.peak_encoder[0], layer_name='weight' ) @@ -70,7 +70,7 @@ def __init__( self.mask_input_weights( peak_tf_mask, - module=self.tf_encoder[0], + module=self.tf_encoder[1], layer_name='weight' )