ddi-fw 0.0.79__tar.gz → 0.0.81__tar.gz

{ddi_fw-0.0.79 → ddi_fw-0.0.81}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ddi_fw
-Version: 0.0.79
+Version: 0.0.81
 Summary: Do not use :)
 Author-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>
 Maintainer-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>

{ddi_fw-0.0.79 → ddi_fw-0.0.81}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ddi_fw"
-version = "0.0.79"
+version = "0.0.81"
 description = "Do not use :)"
 readme = "README.md"
 authors = [

{ddi_fw-0.0.79 → ddi_fw-0.0.81}/src/ddi_fw/ml/ml_helper.py

@@ -2,7 +2,7 @@ from typing import Dict, List, Tuple
 from matplotlib import pyplot as plt
 from ddi_fw.ml.model_wrapper import Result
 from ddi_fw.ml.pytorch_wrapper import PTModelWrapper
-from ddi_fw.experiments.ml_tf import TFModelWrapper
+from ddi_fw.ml.tensorflow_wrapper import TFModelWrapper
 import tensorflow as tf
 from tensorflow import keras
 from keras.models import Model, Sequential
@@ -28,10 +28,6 @@ tf.random.set_seed(1)
 np.random.seed(2)
 np.set_printoptions(precision=4)
 
-
-
- 
-
 class MultiModalRunner:
     # todo model related parameters to config
     def __init__(self, library ,model_func, batch_size=128, epochs=100):

{ddi_fw-0.0.79 → ddi_fw-0.0.81}/src/ddi_fw/pipeline/multi_pipeline.py

@@ -1,3 +1,4 @@
+from collections import defaultdict
 import json
 from pipeline import Pipeline
 import importlib
@@ -43,6 +44,7 @@ class MultiPipeline():
     def __init__(self, experiments_config_file):
         self.experiments_config = load_config(experiments_config_file)
         self.items = []
+        self.pipeline_resuts = dict()
 
     def __create_pipeline(self, config):
         library = config["library"]
@@ -108,4 +110,8 @@ class MultiPipeline():
             batch_size = item['batch_size']
             epochs = item['epochs']
             pipeline.build()
-            pipeline.run(model_type, epochs=epochs, batch_size=batch_size)
+            result = pipeline.run(model_type, epochs=epochs, batch_size=batch_size)
+            self.pipeline_resuts[item['name']] = result
+
+    def results(self):
+        return self.pipeline_resuts

{ddi_fw-0.0.79 → ddi_fw-0.0.81}/src/ddi_fw/pipeline/pipeline.py

@@ -7,7 +7,7 @@ from ddi_fw.langchain.embeddings import PoolingStrategy
 from ddi_fw.datasets import BaseDataset, DDIMDLDataset
 from ddi_fw.langchain.embeddings import SumPoolingStrategy
 import mlflow
-from ml import MultiModalRunner
+from ddi_fw.ml import MultiModalRunner
 
 
 class Pipeline:

{ddi_fw-0.0.79 → ddi_fw-0.0.81}/src/ddi_fw.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ddi_fw
-Version: 0.0.79
+Version: 0.0.81
 Summary: Do not use :)
 Author-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>
 Maintainer-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>