news_classifier / README.md
MikhailRepkin's picture
Update README.md
4108812 verified
|
raw
history blame
1.27 kB
metadata
license: mit
language:
  - ru
metrics:
  - accuracy
pipeline_tag: text-classification
widget:
  - text: >-
      Взрыв газа произошел в 2-этажном доме в поселке под Казанью, пострадали
      четыре человека, сообщает МЧС
    example_title: Новость
  - text: Сын поздравил меня с днём рождения стихами ❤️
    example_title: Не новость

Model Details

Model Description

News_classifier is a fine-tuned model designed for classifying news posts from various russian-language Telegram channels. This model can be integrated into a news aggregation service.

  • Model type: Sentence RuBERT (Russian, cased, 12-layer, 768-hidden, 12-heads, 180M parameters)
  • Language(s) (NLP): russian (ru)
  • License: mit
  • Finetuned from model: DeepPavlov/rubert-base-cased-sentence

Dataset

Russian telegram posts train/valid/test: 2970/165/165

Training Details

  • token max length: 512
  • num labels: 2
  • batch size: 16
  • learning_rate: 2e-5
  • train_epochs: 20
  • weight decay: 0.01

Metrics:

  • Matthews_correlation (training evaluation metric): 0.89
  • Accuracy: 0.95

Label Scheme

  • LABEL_1 - news
  • LABEL_0 - not news