Views
No views yet
SAFE 或 UNSAFE。1import joblib
2
3model = joblib.load('gtc_guard0_model.pkl')
4vectorizer = joblib.load('gtc_guard0_vectorizer.pkl')
5
6def predict(text):
7 vec = vectorizer.transform([text])
8 prob = model.predict_proba(vec)[0, 1]
9 return 'UNSAFE' if prob > 0.5 else 'SAFE'
10
11print(predict("hello")) # SAFE
12print(predict("i will kill you")) # UNSAFE