Ошибка в прогнозировании стратегии в обученной модели ⇐ Python

[Anonymous]

Я пытаюсь запустить программу ML, чтобы получить стратегию исправления на основе ключей ошибок. Я попытался манипулировать тестированием и случайно, но не повезло. Образец правил в «error_strategy_data.csv» < /p>
Примечание. Ниже форматируется в рядах и столбцах «Errormessage», «Стратегия» в CSV < /p>
error_message strategy
Deadlock found when trying to get lock Verify DB network settings and credentials
Syntax error near SELECT Fix SQL syntax or use parameterized queries
Connection refused to database server Retry transaction or redesign locking
Timeout occurred while executing query Retry the request with exponential backoff
< /code>
ниже моя обучающая модель < /p>
import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.pipeline import Pipeline
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.linear_model import LogisticRegression
from sklearn.metrics import classification_report
import joblib

data = pd.read_csv("C:\\Users\\Abhishek.mishra01\\Desktop\\error_strategy_data.csv")
X = data["error_message"]
y = data["strategy"]

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.5, random_state=42)

pipeline = Pipeline([
('tfidf', TfidfVectorizer(
lowercase=True,
stop_words='english',
ngram_range=(1,2),
max_features=5000,
#strip_accents='unicode'
)),
('clf', LogisticRegression(max_iter=1000, C=1))
])

pipeline.fit(X_train, y_train)

preds = pipeline.predict(X_test)
print(classification_report(y_test, preds))

joblib.dump(pipeline, "error_strategy_model.joblib")
< /code>
Мой тестовый класс < /p>
import joblib

# Load
model = joblib.load("error_strategy_model.joblib")

# Predict
error_message = "select"
predicted_strategy = model.predict([error_message])[0]
print("Suggested strategy:", predicted_strategy)
< /code>
Вывод ожидается: < /p>
Fix SQL syntax or use parameterized queries
< /code>
Но получение < /p>
Verify DB network settings and credentials
< /code>
Более интересно, он выбирает только третью строку в качестве стратегии для любого из введенного сообщения об ошибке. Что я могу попробовать дальше?


Подробнее здесь: https://stackoverflow.com/questions/797 ... ined-model