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

[Anonymous]

Я пытаюсь запустить программу ML, чтобы получить стратегию исправления, основанную на клавишах ошибок.
Я попытался манипулировать TestSize и Random Startate, но не повезло.
Sample of Rules в «error_strategy_data.csv» < /p>
Примечание: ниже формат в Rows and Columns "rrormess", "в парнере" /v -vempess "," /v -vempess "," в пар. />errorMessage : Deadlock found when trying to get lock.
strategy: Retry transaction or redesign locking

errorMessage :Syntax error near SELECT
strategy: Fix SQL syntax or use parameterized queries

errorMessage :Connection refused to database server
strategy: Verify DB network settings and credentials

errorMessage :Timeout occurred while executing query
strategy: 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>
Более интересно, он выбирает 3 -й строк в качестве стратегии для любого из введенного сообщения об ошибке.
Пожалуйста, помогите мне < /p>

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