diff --git a/src/rail/estimation/algos/minisom_som.py b/src/rail/estimation/algos/minisom_som.py index 79feaa9..221d944 100644 --- a/src/rail/estimation/algos/minisom_som.py +++ b/src/rail/estimation/algos/minisom_som.py @@ -224,6 +224,7 @@ def __init__(self, args, comm=None): self.n_dim = None def run(self): + self.open_model(**self.config) self.som = self.model['som'] self.usecols = self.model['usecols'] self.column_usage = self.model['column_usage'] diff --git a/src/rail/estimation/algos/somoclu_som.py b/src/rail/estimation/algos/somoclu_som.py index 7430d13..4811f28 100644 --- a/src/rail/estimation/algos/somoclu_som.py +++ b/src/rail/estimation/algos/somoclu_som.py @@ -365,6 +365,7 @@ def get_som_coordinates(self, data, weight_col): return som_coords def run(self): + self.open_model(**self.config) self.som = self.model['som'] self.usecols = self.model['usecols'] self.column_usage = self.model['column_usage']