CyberPeace-Institute
/

Cybersecurity-Knowledge-Graph

Token Classification

Model card Files Files and versions

cpi-connect commited on Sep 27, 2023

Commit

1fb2ae2

·

1 Parent(s): 77eacb7

Upload model

Files changed (1) hide show

model.py +1 -1

model.py CHANGED Viewed

@@ -5,7 +5,7 @@ from cybersecurity_knowledge_graph.nugget_model_utils import CustomRobertaWithPO
 from cybersecurity_knowledge_graph.args_model_utils import CustomRobertaWithPOS as ArgumentModel
 from cybersecurity_knowledge_graph.realis_model_utils import CustomRobertaWithPOS as RealisModel
-from cybersecurity_knowledge_graph.configuration import CybersecurityKnowledgeGraphConfig
 from cybersecurity_knowledge_graph.event_nugget_predict import create_dataloader as event_nugget_dataloader
 from cybersecurity_knowledge_graph.event_realis_predict import create_dataloader as event_realis_dataloader

 from cybersecurity_knowledge_graph.args_model_utils import CustomRobertaWithPOS as ArgumentModel
 from cybersecurity_knowledge_graph.realis_model_utils import CustomRobertaWithPOS as RealisModel
+from .configuration import CybersecurityKnowledgeGraphConfig
 from cybersecurity_knowledge_graph.event_nugget_predict import create_dataloader as event_nugget_dataloader
 from cybersecurity_knowledge_graph.event_realis_predict import create_dataloader as event_realis_dataloader