Edit model card

Instruction Detection Model

Welcome to the repository for the instruction detection model! This model, hosted on the Hugging Face Model Hub, is designed specifically for detecting instructions in newspaper titles. Ukranian Language Only. Utilizing token classification, it scans through the given input - a newspaper title, and labels tokens that appear to signify an instruction.

Model Card: zeusfsx/instruction-detection

Table of Contents

Introduction

In the age of information, newspaper titles are often crafted to attract attention and occasionally incorporate direct or indirect instructions. This model can help analyze these titles, detect such instructions, and tag them accordingly.

It employs token classification task, a common technique in Natural Language Processing (NLP), to detect and label instructions in the text.

Usage

Here's how to use this model:

In Python

from transformers import AutoTokenizer, AutoModelForTokenClassification
from transformers import pipeline

tokenizer = AutoTokenizer.from_pretrained("zeusfsx/instruction-detection")
model = AutoModelForTokenClassification.from_pretrained("zeusfsx/instruction-detection")

nlp = pipeline("ner", model=model, tokenizer=tokenizer)
example = "Your example newspaper title here"

output = nlp(example)
print(output)

This will return a list of recognized tokens marked with label 'INSTRUCTION'.

Training

It's based on the transformer architecture and specifically uses the xlm-roberta-base-uk model from ukr-models, fine-tuned for the token classification task. The training data was carefully chosen to include a balanced distribution of titles containing instructions and those not containing instructions. The dataset contains newspaper titles (~6k titles), with tokens representing instructions manually labeled.

Evaluation

Model performance was evaluated using a held-out test set, again consisting of manually labeled newspaper titles. F1 - 0.9601, ACCURACY - 0.9968 for the 'INSTRUCTION' label

License

This project is licensed under the terms of the MIT license.

Citation

If you use our model or this repository in your research, please cite it as follows:

@misc{instruction-detection,
  author = {Oleksandr Korovii},
  title = {Instruction Detection Model},
  year = {2023},
  publisher = {HuggingFace Model Hub},
  url = {https://huggingface.co/zeusfsx/instruction-detection},
  note = {Accessed: 2023-06-20}
}

Contact Information

For any questions or suggestions, feel free to open an issue in this repository. Contributions to improve this model or the associated documentation are welcome!

Downloads last month
1
Safetensors
Model size
109M params
Tensor type
F32
·