wbrooks commited on
Commit
9fbd1cf
·
1 Parent(s): 234c1f5

being more selective with the columns that print in the output

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -44,8 +44,9 @@ def search(q: str = Query(..., description="Search query")):
44
 
45
  res_combined = joined.with_columns(
46
  (0.7 * pl.col("rank-sbert") + 0.3 * pl.col("rank-tfidf")).alias("rank-combined"),
47
- pl.col("file").str.strip_prefix(path_prefix).alias("file")
48
- ).sort("rank-combined")
 
49
 
50
  #return {"request": request, "results": str(res_combined)}
51
  #return {"request": request, "results": res_combined.to_dicts()}
 
44
 
45
  res_combined = joined.with_columns(
46
  (0.7 * pl.col("rank-sbert") + 0.3 * pl.col("rank-tfidf")).alias("rank-combined"),
47
+ pl.col("file").str.strip_prefix(path_prefix).alias("file"),
48
+ (20.0 / pl.col('rank-combined')).round(2).alias('confidence')
49
+ ).sort("rank-combined").select(['file', 'confidence'])
50
 
51
  #return {"request": request, "results": str(res_combined)}
52
  #return {"request": request, "results": res_combined.to_dicts()}