CyberPeace-Institute
/

Cybersecurity-Knowledge-Graph

Token Classification

Model card Files Files and versions

cpi-connect commited on Sep 28, 2023

Commit

44c6516

·

1 Parent(s): acd5094

Upload model

Files changed (1) hide show

configuration.py +5 -5

configuration.py CHANGED Viewed

@@ -1,7 +1,7 @@
 from transformers import PretrainedConfig
 import torch
-import cybersecurity_knowledge_graph.utils
@@ -17,9 +17,9 @@ class CybersecurityKnowledgeGraphConfig(PretrainedConfig):
         self.event_argument_model_path = event_argument_model_path
         self.event_realis_model_path = event_realis_model_path
-        self.event_nugget_list = cybersecurity_knowledge_graph.utils.event_nugget_list
-        self.event_args_list = cybersecurity_knowledge_graph.utils.event_args_list
-        self.realis_list = cybersecurity_knowledge_graph.utils.realis_list
-        self.arg_2_role = cybersecurity_knowledge_graph.utils.arg_2_role
         super().__init__(**kwargs)

 from transformers import PretrainedConfig
 import torch
+from . import utils
         self.event_argument_model_path = event_argument_model_path
         self.event_realis_model_path = event_realis_model_path
+        self.event_nugget_list = utils.event_nugget_list
+        self.event_args_list = utils.event_args_list
+        self.realis_list = utils.realis_list
+        self.arg_2_role = utils.arg_2_role
         super().__init__(**kwargs)