From 940f7aa1b5447b79cc39e87171445e6bd0c219e3 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 1 Sep 2022 16:20:23 +0100 Subject: [PATCH] ai: set self.model --- aimodel/src/lib/ai/RainfallWaterContraster.py | 3 +-- aimodel/src/lib/ai/model_rainfallwater_contrastive.py | 4 +++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/aimodel/src/lib/ai/RainfallWaterContraster.py b/aimodel/src/lib/ai/RainfallWaterContraster.py index b452799..178bcae 100644 --- a/aimodel/src/lib/ai/RainfallWaterContraster.py +++ b/aimodel/src/lib/ai/RainfallWaterContraster.py @@ -57,8 +57,7 @@ class RainfallWaterContraster(object): def make_model(self): - model = model_rainfallwater_contrastive(batch_size=self.batch_size, **self.kwargs) - return model + return model_rainfallwater_contrastive(batch_size=self.batch_size, **self.kwargs) def load_model(self, filepath_checkpoint): diff --git a/aimodel/src/lib/ai/model_rainfallwater_contrastive.py b/aimodel/src/lib/ai/model_rainfallwater_contrastive.py index 23b7a75..f607506 100644 --- a/aimodel/src/lib/ai/model_rainfallwater_contrastive.py +++ b/aimodel/src/lib/ai/model_rainfallwater_contrastive.py @@ -49,4 +49,6 @@ def model_rainfallwater_contrastive(shape_rainfall, shape_water, batch_size=64, model.compile( optimizer="Adam", loss=LossContrastive(batch_size=batch_size, weight_temperature=weight_temperature) - ) \ No newline at end of file + ) + + return model \ No newline at end of file