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

[Anonymous]

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

стратегия: трантриация или блокировка редизайна
errormessage: ошибка синтаксиса рядом с Select

стратегия: исправление SQL Syntax или использование параметризованных запросов
rerrormessage: connection defressed database server

rerrormessage: armentaintive: armentaintive d /> Errormessage: Тайм-аут произошел при выполнении запроса

стратегия: повторно запрос с экспоненциальным отбором < /p>
Ниже моя обучающая модель < /p>
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") ```

My Test class

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)

Output expected:
"Fix SQL syntax or use parameterized queries" but getting "Verify DB network settings and credentials"

More interestingly, it is only picking the 3rd row as a strategy for any of the error message entered.
Please help me out



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