From 74504aef62d1584df893087d6f73fab4d556d881 Mon Sep 17 00:00:00 2001 From: Eric Charles Date: Tue, 15 Aug 2023 15:01:36 -0700 Subject: [PATCH] move open_model from init to run --- src/rail/estimation/algos/minisom_som.py | 1 + src/rail/estimation/algos/somoclu_som.py | 1 + 2 files changed, 2 insertions(+) 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']