Senior
Ai un set mic de date cu recenzii de filme în rusă și eticheta lor binară de tonalitate (etichete: 0 — negativ, 1 — pozitiv). Sarcina ta este să implementezi un pipeline complet în 15 minute pentru a antrena un model simplu de clasificare a textului fără a folosi encodere pre-antrenate precum BERT, și să afișezi acuratețea (accuracy) pe aceste date. reviews = [ "Acest film este pur și simplu incredibil!", "Joc de actorii oribil și poveste plictisitoare.", "Mi-a plăcut fiecare moment!", "Cel mai rău film pe care l-am văzut.", "Lucru excelent de regie și poveste minunată.", "Insuportabil de plictisitor și previzibil." ] labels = [1, 0, 1, 0, 1, 0]
sobes.tech AI
Răspuns de la AI
Для быстрого решения задачи классификации текстов на русском языке без использования готовых энкодеров можно сделать следующий pipeline:
- Токенизация и простая предобработка (приведение к нижнему регистру, удаление знаков препинания).
- Векторизация текста с помощью мешка слов (Bag of Words) с помощью sklearn CountVectorizer.
- Обучение простой модели, например, логистической регрессии.
- Оценка точности на тех же данных (так как датасет очень маленький).
Пример кода на 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 минут получить базовую модель классификации с приемлемой точностью на маленьком датасете.