Yehor commited on
Commit
9581c74
·
1 Parent(s): ff99877

Use all metrics

Browse files
Files changed (1) hide show
  1. app.py +15 -10
app.py CHANGED
@@ -64,7 +64,7 @@ tech_libraries = f"""
64
  - evaluate: {version("evaluate")}
65
  - pandas: {version("pandas")}
66
  - polars: {version("polars")}
67
- - polars_distance: {version("polars_distance")}
68
  """.strip()
69
 
70
 
@@ -160,23 +160,28 @@ def process_file(file_name, _batch_mode, _calculate_distance, _calculate_metrics
160
  ]
161
 
162
  df = pl.DataFrame(df_pd)
163
- elif _calculate_distance:
 
 
 
 
 
 
 
 
164
  df = df.with_columns(
165
  pld.col("prediction").dist_str.levenshtein("reference").alias("distance")
166
  )
167
 
 
168
  fields = [
169
  "distance",
170
- "prediction",
171
- "reference",
172
- ]
173
- else:
174
- fields = [
175
- "prediction",
176
- "reference",
177
  ]
178
 
179
- return df.select(fields)
 
 
180
 
181
 
182
  demo = gr.Blocks(
 
64
  - evaluate: {version("evaluate")}
65
  - pandas: {version("pandas")}
66
  - polars: {version("polars")}
67
+ - polars-distance: {version("polars_distance")}
68
  """.strip()
69
 
70
 
 
160
  ]
161
 
162
  df = pl.DataFrame(df_pd)
163
+ else:
164
+ fields = [
165
+ "prediction",
166
+ "reference",
167
+ ]
168
+
169
+ df = df.select(fields)
170
+
171
+ if _calculate_distance:
172
  df = df.with_columns(
173
  pld.col("prediction").dist_str.levenshtein("reference").alias("distance")
174
  )
175
 
176
+ # add distance to the first position
177
  fields = [
178
  "distance",
179
+ *fields,
 
 
 
 
 
 
180
  ]
181
 
182
+ df = df.select(fields)
183
+
184
+ return df
185
 
186
 
187
  demo = gr.Blocks(