ruSpam_big / README.md
darkQibit's picture
Update README.md
3300bb2 verified
metadata
library_name: transformers
license: cc-by-nc-nd-4.0
datasets:
  - ruSpamModels/russian-spam-detection

ruSpam_big

Модель ruSpam_big — нейросеть для определения спама в русскоязычных текстах. Обучена на почти 5 миллионах примеров, показывает высокую точность и устойчивость, но иногда может ошибаться на сообщениях с коммерческой тематикой.

Архитектура

  • Базовая модель: RuBERT
  • Тип задачи: бинарная классификация (СПАМ / НЕ СПАМ)
  • Количество меток: 1 (используется сигмоида для вероятности спама)
  • Фреймворк: PyTorch + Transformers (Hugging Face)

Пример использования

import torch
from transformers import AutoTokenizer, AutoModelForSequenceClassification

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
tokenizer = AutoTokenizer.from_pretrained("ruSpamModels/ruSpam_big")
model = AutoModelForSequenceClassification.from_pretrained(
    "NeuroSpaceX/ruSpam_big", num_labels=1
).to(device).eval()

text = "Пример сообщения"
encoding = tokenizer(text, padding="max_length", truncation=True, max_length=512, return_tensors="pt")
input_ids = encoding["input_ids"].to(device)
attention_mask = encoding["attention_mask"].to(device)

with torch.no_grad():
    outputs = model(input_ids, attention_mask=attention_mask).logits
    prediction = torch.sigmoid(outputs).cpu().numpy()[0][0]
    is_spam = prediction >= 0.5

print(f"Результат: {'СПАМ' if is_spam else 'НЕ СПАМ'} (уверенность: {prediction:.4f})")

💖 Поддержка проекта

Если проект оказался полезен, вы можете поддержать его развитие:

  • TON:
    UQAvc2APxFcmNg0-K8TJ8ykdkl2GjwveuVfFd8-NOwHGEeqv

  • Рубли (Telegram):
    Перейдите по старт-ссылке бота
    👉 @ModProtectorBot

Спасибо за поддержку ❤️