Sobes.tech
Senior

Имаш малък набор от данни с рецензии за филми на руски език и техните бинарни етикети за тоналност (етикети: 0 — негативен, 1 — позитивен). Задачата ти е да реализираш пълен pipeline за обучение на прост модел за класификация на текст без използване на готови енкодери като BERT за 15 минути и да покажеш точността (accuracy) върху тези данни. reviews = [ "Този филм е просто невероятен!", "Ужасна игра на актьорите и скучен сюжет.", "Много ми хареса всеки момент!", "Най-лошият филм, който съм виждал.", "Отлична режисура и чудесна история.", "Невъзможно скучно и предсказуемо." ] labels = [1, 0, 1, 0, 1, 0]

sobes.tech AI

Отговор от AI

Для быстрого решения задачи классификации текстов на русском языке без использования готовых энкодеров можно сделать следующий pipeline:

  1. Токенизация и простая предобработка (приведение к нижнему регистру, удаление знаков препинания).
  2. Векторизация текста с помощью мешка слов (Bag of Words) с помощью sklearn CountVectorizer.
  3. Обучение простой модели, например, логистической регрессии.
  4. Оценка точности на тех же данных (так как датасет очень маленький).

Пример кода на Python:

from sklearn.feature_extraction.text import CountVectorizer
from sklearn.linear_model import LogisticRegression
from sklearn.metrics import accuracy_score

reviews = [
    "Этот фильм просто потрясающий!",
    "Ужасная игра актёров и скучный сюжет.",
    "Мне очень понравилось каждое мгновение!",
    "Самый худший фильм, который я видел.",
    "Отличная операторская работа и замечательная история.",
    "Невыносимо скучно и предсказуемо."
]
labels = [1, 0, 1, 0, 1, 0]

# Векторизация
vectorizer = CountVectorizer(lowercase=True)
X = vectorizer.fit_transform(reviews)

# Обучение модели
model = LogisticRegression()
model.fit(X, labels)

# Предсказание и оценка
preds = model.predict(X)
accuracy = accuracy_score(labels, preds)
print(f"Accuracy: {accuracy:.2f}")

Такой подход позволит за 15 минут получить базовую модель классификации с приемлемой точностью на маленьком датасете.