ddi-fw 0.0.55__py3-none-any.whl → 0.0.56__py3-none-any.whl

ddi_fw/experiments/pipeline_ner.py

@@ -23,6 +23,7 @@ class NerParameterSearch:
                  experiment_tags,
                  tracking_uri,
                  dataset_type: BaseDataset,
+                 columns:list,
                  umls_code_types: List[UMLSCodeTypes],
                  text_types=List[DrugBankTextDataTypes],
                  min_threshold_dict: Dict[str, float] = defaultdict(float),
@@ -34,6 +35,7 @@ class NerParameterSearch:
         self.tracking_uri = tracking_uri
 
         self.dataset_type = dataset_type
+        self.columns = columns
         self.umls_code_types = umls_code_types
         self.text_types = text_types
         self.min_threshold_dict = min_threshold_dict
@@ -43,17 +45,17 @@ class NerParameterSearch:
     def build(self):
         self.datasets = {}
         self.items = []
-        columns = ['tui', 'cui', 'entities']
+        # columns = ['tui', 'cui', 'entities']
         if self.umls_code_types is not None and self.text_types is not None:
             # add checking statements
             _umls_codes = [t.value[0] for t in self.umls_code_types]
             _text_types = [t.value[0] for t in self.text_types]
             _columns = [f'{item[0]}_{item[1]}' for item in product(
                 _umls_codes, _text_types)]
-            columns.extend(_columns)
-        print(f'Columns: {columns}')
+            self.columns.extend(_columns)
+        print(f'Columns: {self.columns}')
         self.ner_df = CTakesNER().load(filename=self.ner_data_file)  if self.ner_data_file else None
-        for column in columns:
+        for column in self.columns:
             min_threshold = self.min_threshold_dict[column]
             max_threshold = self.max_threshold_dict[column]
             kwargs = {}

{ddi_fw-0.0.55.dist-info → ddi_fw-0.0.56.dist-info}/METADATA

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

{ddi_fw-0.0.55.dist-info → ddi_fw-0.0.56.dist-info}/RECORD

@@ -61,7 +61,7 @@ ddi_fw/experiments/custom_torch_model.py,sha256=iQ_R_EApzD2JCcASN8cie6D21oh7VCxa
 ddi_fw/experiments/evaluation_helper.py,sha256=pY69cezV3WzrXw1bduIwRJfah1w3wXJ2YyTNim1J7ko,9349
 ddi_fw/experiments/pipeline.py,sha256=wttkvdzGP9d3jC9nx2iZul4hbogXkRho6eDns0yfLiE,5380
 ddi_fw/experiments/pipeline_builder_pattern.py,sha256=q1PNEQFoO5U3UidEoGB8rgLA7KXr4FsJTXEug5c5UJg,5466
-ddi_fw/experiments/pipeline_ner.py,sha256=JERKAaPdgKt2wjfVauOd3HXOGbLLoYLNxNCAv9CO_vg,4757
+ddi_fw/experiments/pipeline_ner.py,sha256=g-Tp7P7hmFxOvuqBV_Cfxt-4xe6xft9WC43S3NFns2Y,4838
 ddi_fw/experiments/tensorflow_helper.py,sha256=Y-gD9qyqFFPl6HAvM_tIa5Y6em2YmafPCL1KMrK6eb8,11768
 ddi_fw/experiments/test.py,sha256=z1TfBpK75zGKpp2ZU8f6APjZlgBFthaCBN61YB9ma4o,2049
 ddi_fw/ner/__init__.py,sha256=JwhGXrepomxPSsGsg2b_xPRC72AjvxOIn2CW5Mvscn0,26
@@ -83,7 +83,7 @@ ddi_fw/utils/enums.py,sha256=19eJ3fX5eRK_xPvkYcukmug144jXPH4X9zQqtsFBj5A,671
 ddi_fw/utils/py7zr_helper.py,sha256=dgfHqXDBWys1hmd1JlHhYyZGxrzYWi6siYiUq3bnLuI,4698
 ddi_fw/utils/utils.py,sha256=szwnxMTDRrZoeNRyDuf3aCbtzriwtaRk4mHSH3asLdA,4301
 ddi_fw/utils/zip_helper.py,sha256=YRZA4tKZVBJwGQM0_WK6L-y5MoqkKoC-nXuuHK6CU9I,5567
-ddi_fw-0.0.55.dist-info/METADATA,sha256=736seAJsPdjZQPhFly5pkPPGi7SMWr6XqNgUKKRhC2I,1565
-ddi_fw-0.0.55.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-ddi_fw-0.0.55.dist-info/top_level.txt,sha256=PMwHICFZTZtcpzQNPV4UQnfNXYIeLR_Ste-Wfc1h810,7
-ddi_fw-0.0.55.dist-info/RECORD,,
+ddi_fw-0.0.56.dist-info/METADATA,sha256=iOaA6X5rIXTUxkNeS0JGXTjvegkL26R6-taf8bRiyIQ,1565
+ddi_fw-0.0.56.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+ddi_fw-0.0.56.dist-info/top_level.txt,sha256=PMwHICFZTZtcpzQNPV4UQnfNXYIeLR_Ste-Wfc1h810,7
+ddi_fw-0.0.56.dist-info/RECORD,,