hysts HF staff commited on
Commit
8892a39
·
1 Parent(s): db717e5

Fix filters

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -88,7 +88,7 @@ def filter_models(
88
  print(f"After type filter: {filtered_df.shape}")
89
 
90
  # Precision フィルタリング
91
- filtered_df = filtered_df[filtered_df["Precision"].isin(precision_query + ["Unknown", "?"])]
92
  print(f"After precision filter: {filtered_df.shape}")
93
 
94
  # Model Size フィルタリング
@@ -101,13 +101,11 @@ def filter_models(
101
  print(f"After size filter: {filtered_df.shape}")
102
 
103
  # Add Special Tokens フィルタリング
104
- filtered_df = filtered_df[filtered_df["Add Special Tokens"].isin(add_special_tokens_query + ["Unknown", "?"])]
105
  print(f"After add_special_tokens filter: {filtered_df.shape}")
106
 
107
  # Num Few Shots フィルタリング
108
- filtered_df = filtered_df[
109
- filtered_df["Few-shot"].astype(str).isin([str(x) for x in num_few_shots_query] + ["Unknown", "?"])
110
- ]
111
  print(f"After num_few_shots filter: {filtered_df.shape}")
112
 
113
  print("Filtered dataframe head:")
 
88
  print(f"After type filter: {filtered_df.shape}")
89
 
90
  # Precision フィルタリング
91
+ filtered_df = filtered_df[filtered_df["Precision"].isin(precision_query)]
92
  print(f"After precision filter: {filtered_df.shape}")
93
 
94
  # Model Size フィルタリング
 
101
  print(f"After size filter: {filtered_df.shape}")
102
 
103
  # Add Special Tokens フィルタリング
104
+ filtered_df = filtered_df[filtered_df["Add Special Tokens"].isin(add_special_tokens_query)]
105
  print(f"After add_special_tokens filter: {filtered_df.shape}")
106
 
107
  # Num Few Shots フィルタリング
108
+ filtered_df = filtered_df[filtered_df["Few-shot"].astype(str).isin(num_few_shots_query)]
 
 
109
  print(f"After num_few_shots filter: {filtered_df.shape}")
110
 
111
  print("Filtered dataframe head:")