Instructions to use CyberPeace-Institute/Cybersecurity-Knowledge-Graph with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use CyberPeace-Institute/Cybersecurity-Knowledge-Graph with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="CyberPeace-Institute/Cybersecurity-Knowledge-Graph", trust_remote_code=True)# Load model directly from transformers import AutoModelForTokenClassification model = AutoModelForTokenClassification.from_pretrained("CyberPeace-Institute/Cybersecurity-Knowledge-Graph", trust_remote_code=True, dtype="auto") - Notebooks
- Google Colab
- Kaggle
Commit ·
d63bc19
1
Parent(s): 9070e2e
Update README.md
Browse files
README.md
CHANGED
|
@@ -1,8 +1,11 @@
|
|
| 1 |
---
|
| 2 |
language:
|
| 3 |
- en
|
| 4 |
-
library_name: keras
|
| 5 |
pipeline_tag: token-classification
|
| 6 |
tags:
|
| 7 |
- legal
|
| 8 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
language:
|
| 3 |
- en
|
|
|
|
| 4 |
pipeline_tag: token-classification
|
| 5 |
tags:
|
| 6 |
- legal
|
| 7 |
+
---
|
| 8 |
+
|
| 9 |
+
# Knowledge Graph Extraction for Cyber incidents
|
| 10 |
+
This model has been finetuned with SecureBERT (https://arxiv.org/abs/2204.02685)
|
| 11 |
+
on the CASIE dataset (https://ieeexplore.ieee.org/document/9776031). We have implemented the approach described in the CASIE paper.
|