How to use CyberPeace-Institute/SecureBERT-NER with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="CyberPeace-Institute/SecureBERT-NER")
# Load model directly from transformers import AutoTokenizer, AutoModelForTokenClassification tokenizer = AutoTokenizer.from_pretrained("CyberPeace-Institute/SecureBERT-NER") model = AutoModelForTokenClassification.from_pretrained("CyberPeace-Institute/SecureBERT-NER")
Hi.. Is it possible to share the training file that was used? Thanks.
Hello,
You can find the dataset in the following Github page:https://github.com/wangxuren/APTNER
· Sign up or log in to comment
