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

[Anonymous]

Я пытаюсь запустить программу ML, чтобы получить стратегию исправления, основанную на клавишах ошибок.
Я попытался манипулировать TestSize и Random Startate, но не повезло.
Sample of Rules в «error_strategy_data.csv» < /p>
Примечание: ниже формат в Rows and Columns "rrormess", "в парнере" /v -vempess "," /v -vempess "," в пар. />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>
Более интересно, он выбирает 3 -й строк в качестве стратегии для любого из введенного сообщения об ошибке.
Пожалуйста, помогите мне < /p>

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