ddi-fw 0.0.55__tar.gz → 0.0.56__tar.gz

{ddi_fw-0.0.55 → ddi_fw-0.0.56}/PKG-INFO

@@ -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 → ddi_fw-0.0.56}/pyproject.toml

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

{ddi_fw-0.0.55 → ddi_fw-0.0.56}/src/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 → ddi_fw-0.0.56}/src/ddi_fw.egg-info/PKG-INFO

@@ -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>