ipynb中的结果:
array([[0.99998873,0.34232343,0.97835689,0.11738408,0.96085243,0.03861971]])
结果为django:
[[0.07742869 0.00020369 0.00456771 0.00032158 0.00515316 0.00088939]]
views.py的代码如下所示,
vect = TfidfVectorizer(max_features=40000,stop_words='english')
target = ['toxic','severe_toxic','obscene','threat','insult','identity_hate']
data = pd.read_csv('train.csv')
X = data.comment_text
xt = vect.fit_transform(X)
input_comment1 = str(input_comment)
input_comment1 = [input_comment1]
input_comment1 = vect.transform(input_comment1)
load_model = joblib.load('lr.pkl')
output_class = load_model.predict_proba(input_comment1).toarray()
print(output_class)