Christopher Román Jaimes commited on
Commit
1b6b3cf
·
1 Parent(s): 1ef8976

chore: add raw predictions.

Browse files
Files changed (1) hide show
  1. app.py +1 -33
app.py CHANGED
@@ -138,38 +138,6 @@ def clean_prediction(row, feature_name, threshols_dict, clean_functions_dict):
138
  return prediction_clean
139
  else:
140
  return None
141
-
142
- def calculate_metrics(X, feature_name, data_type):
143
- true_positives = 0
144
- true_negatives = 0
145
- false_positives = 0
146
- false_negatives = 0
147
- for pred, true in zip(X[f"clean_pred_{feature_name}"], X[f"clean_{feature_name}"]):
148
- if isinstance(pred, data_type):
149
- if isinstance(true, data_type):
150
- if pred == true:
151
- true_positives += 1
152
- else:
153
- false_positives += 1
154
- else:
155
- false_positives += 1
156
- else:
157
- if isinstance(true, data_type):
158
- false_negatives += 1
159
- else:
160
- true_negatives += 1
161
-
162
- # Calculate Metrics
163
- precision = true_positives / (true_positives + false_positives) if (true_positives + false_positives) != 0 else np.nan
164
- recall = true_positives / (true_positives + false_negatives) if (true_positives + false_negatives) != 0 else np.nan
165
- f1_score = 2 * (precision * recall) / (precision + recall) if (precision + recall) != 0 else np.nan
166
- metrics = {
167
- "precision": precision,
168
- "recall": recall,
169
- "f1_score": f1_score,
170
- }
171
-
172
- return metrics
173
 
174
  def extract_remodeling_year_from_string(string):
175
  if isinstance(string, str):
@@ -251,7 +219,7 @@ def generate_answer(text):
251
 
252
  result_json = json.dumps(entities_cleaned, indent = 4, ensure_ascii = False)
253
 
254
- return result_json
255
 
256
  # Cambiar a entrada de texto
257
  #text_input = gr.inputs.Textbox(lines=15, label="Input Text")
 
138
  return prediction_clean
139
  else:
140
  return None
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
141
 
142
  def extract_remodeling_year_from_string(string):
143
  if isinstance(string, str):
 
219
 
220
  result_json = json.dumps(entities_cleaned, indent = 4, ensure_ascii = False)
221
 
222
+ return result_json + "\n \n" + entities_formatted
223
 
224
  # Cambiar a entrada de texto
225
  #text_input = gr.inputs.Textbox(lines=15, label="Input Text")