820nam commited on
Commit
14ae745
·
verified ·
1 Parent(s): 531e3fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -61,7 +61,7 @@ def initialize_model():
61
  vectorizer = joblib.load("tfidf_vectorizer.pkl")
62
  else:
63
  # 초기 모델 및 벡터라이저 설정
64
- model = SGDClassifier(loss='log', max_iter=5, tol=None) # SGDClassifier를 사용해 Incremental Learning 적용
65
  vectorizer = TfidfVectorizer(max_features=1000, stop_words="english")
66
  return model, vectorizer
67
 
@@ -75,6 +75,7 @@ def incremental_training(texts, labels, model, vectorizer):
75
  joblib.dump(vectorizer, "tfidf_vectorizer.pkl")
76
  return model, vectorizer
77
 
 
78
  # GPT-4를 이용해 반대 관점 기사 생성
79
  def generate_article_gpt4(prompt):
80
  try:
 
61
  vectorizer = joblib.load("tfidf_vectorizer.pkl")
62
  else:
63
  # 초기 모델 및 벡터라이저 설정
64
+ model = SGDClassifier(loss='log_loss', max_iter=5, tol=None) # 'log_loss'로 수정
65
  vectorizer = TfidfVectorizer(max_features=1000, stop_words="english")
66
  return model, vectorizer
67
 
 
75
  joblib.dump(vectorizer, "tfidf_vectorizer.pkl")
76
  return model, vectorizer
77
 
78
+
79
  # GPT-4를 이용해 반대 관점 기사 생성
80
  def generate_article_gpt4(prompt):
81
  try: