johannes-garstenauer/structs_token_size_4_reduced_labelled_eval
Viewer • Updated • 269k • 28 • 1
How to use johannes-garstenauer/distilbert_class_heaps with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="johannes-garstenauer/distilbert_class_heaps") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("johannes-garstenauer/distilbert_class_heaps")
model = AutoModelForSequenceClassification.from_pretrained("johannes-garstenauer/distilbert_class_heaps")DistilBERT for sequence classification trained on OpenSSH heap data structures dataset for the purpose of generating representations. This model was created for the thesis "Generating Robust Representations of Structures in OpenSSH Heap Dumps" by Johannes Garstenauer. It is finetuned from "johannes-garstenauer/distilbert_masking_heaps".
Training data: https://huggingface.co/datasets/johannes-garstenauer/structs_token_size_4_reduced_labelled_train Validation data: https://huggingface.co/datasets/johannes-garstenauer/structs_token_size_4_reduced_labelled_eval