pentarosarium commited on
Commit
f8fa791
·
1 Parent(s): c8cc09f

progress more 38+

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -29,7 +29,7 @@ def create_analysis_data(df):
29
  analysis_data = []
30
  for _, row in df.iterrows():
31
  if any(row[model] == 'Negative' for model in ['FinBERT', 'RoBERTa', 'FinBERT-Tone']):
32
- analysis_data.append([row['Объект'], 'РИСК УБЫТКА', row['Заголовок'], row['Выдержки из текста']])
33
  return pd.DataFrame(analysis_data, columns=['Объект', 'Тип риска', 'Заголовок', 'Текст'])
34
 
35
  # Function for lemmatizing Russian text
@@ -201,11 +201,12 @@ def create_output_file(df, uploaded_file, analysis_df):
201
  ws.cell(row=r_idx, column=c_idx, value=value)
202
 
203
  # Process data for 'Значимые' sheet
 
204
  significant_data = []
205
  for _, row in df.iterrows():
206
  if any(row[model] in ['Negative', 'Positive'] for model in ['FinBERT', 'RoBERTa', 'FinBERT-Tone']):
207
  sentiment = 'Negative' if any(row[model] == 'Negative' for model in ['FinBERT', 'RoBERTa', 'FinBERT-Tone']) else 'Positive'
208
- significant_data.append([row['Объект'], sentiment, row['Заголовок'], row['Выдержки из текста']])
209
 
210
  # Write 'Значимые' sheet
211
  ws = wb['Значимые']
 
29
  analysis_data = []
30
  for _, row in df.iterrows():
31
  if any(row[model] == 'Negative' for model in ['FinBERT', 'RoBERTa', 'FinBERT-Tone']):
32
+ analysis_data.append([row['Объект'], row['Заголовок'], 'РИСК УБЫТКА', '', row['Выдержки из текста']])
33
  return pd.DataFrame(analysis_data, columns=['Объект', 'Тип риска', 'Заголовок', 'Текст'])
34
 
35
  # Function for lemmatizing Russian text
 
201
  ws.cell(row=r_idx, column=c_idx, value=value)
202
 
203
  # Process data for 'Значимые' sheet
204
+
205
  significant_data = []
206
  for _, row in df.iterrows():
207
  if any(row[model] in ['Negative', 'Positive'] for model in ['FinBERT', 'RoBERTa', 'FinBERT-Tone']):
208
  sentiment = 'Negative' if any(row[model] == 'Negative' for model in ['FinBERT', 'RoBERTa', 'FinBERT-Tone']) else 'Positive'
209
+ significant_data.append([row['Объект'], '', sentiment, '', row['Заголовок'], row['Выдержки из текста']])
210
 
211
  # Write 'Значимые' sheet
212
  ws = wb['Значимые']