Downloads · 30 days
0
hadikhamoud/DRU-RE-one-hot-no-dropout
DRU-RE-one-hot-no-dropout is a machine learning model from hadikhamoud. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Fine-tuned relation extraction model using NeoAraBERT with typed subject/object markers and directional entity-type features.
Downloads · 30 days
0
Access
Public
Updated Jul 9, 2026
Repo size
7 GB
Likes
0
Public
Click a slice to open those files.
.pt4 GB · 44%
From the Hugging Face model README
Fine-tuned relation extraction model using NeoAraBERT with typed subject/object markers and directional entity-type features.
U4RASD/NeoAraBERT
Input sentence is marked with typed subject/object markers, encoded using NeoAraBERT, then the CLS vector is concatenated with a 42-dimensional subject/object type feature vector.
Final classifier input:
CLS(768) + type_features(42) = 810
Output labels:
41 relation classes.
model/pytorch_model.bin: custom PyTorch state dictmodel/: tokenizer and encoder configconfigs/labels.jsonconfigs/entity_types.jsonconfigs/special_marker_tokens.jsonconfigs/run_config.jsonartifacts/: metrics and prediction files if generatedThis is a custom nn.Module, not a plain AutoModelForSequenceClassification.
To reload it, recreate the same model class and load model/pytorch_model.bin.