n1ck-guo commited on
Commit
46ccd73
1 Parent(s): 26202c6

Signed-off-by: n1ck-guo <heng.guo@intel.com>

Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -386,17 +386,22 @@ def filter_models(
386
 
387
  filtered_df = filtered_df.loc[df[AutoEvalColumn.group_size.name].isin(group_dtype)]
388
 
389
- print(filtered_df['model_name_for_query'])
390
 
391
  numeric_interval = pd.IntervalIndex(sorted([NUMERIC_INTERVALS[s] for s in size_query]))
392
  params_column = pd.to_numeric(df[AutoEvalColumn.params.name], errors="coerce")
393
  mask = params_column.apply(lambda x: any(numeric_interval.contains(x)))
394
  filtered_df = filtered_df.loc[mask]
 
 
 
395
 
396
  numeric_interval_params = pd.IntervalIndex(sorted([NUMERIC_MODELSIZE[s] for s in params_query]))
397
  params_column_params = pd.to_numeric(df[AutoEvalColumn.model_size.name], errors="coerce")
398
  mask_params = params_column_params.apply(lambda x: any(numeric_interval_params.contains(x)))
399
  filtered_df = filtered_df.loc[mask_params]
 
 
400
 
401
  return filtered_df
402
 
 
386
 
387
  filtered_df = filtered_df.loc[df[AutoEvalColumn.group_size.name].isin(group_dtype)]
388
 
389
+ print(filtered_df['model_name_for_query'].to_list())
390
 
391
  numeric_interval = pd.IntervalIndex(sorted([NUMERIC_INTERVALS[s] for s in size_query]))
392
  params_column = pd.to_numeric(df[AutoEvalColumn.params.name], errors="coerce")
393
  mask = params_column.apply(lambda x: any(numeric_interval.contains(x)))
394
  filtered_df = filtered_df.loc[mask]
395
+
396
+ print('step 2')
397
+ print(filtered_df['model_name_for_query'].to_list())
398
 
399
  numeric_interval_params = pd.IntervalIndex(sorted([NUMERIC_MODELSIZE[s] for s in params_query]))
400
  params_column_params = pd.to_numeric(df[AutoEvalColumn.model_size.name], errors="coerce")
401
  mask_params = params_column_params.apply(lambda x: any(numeric_interval_params.contains(x)))
402
  filtered_df = filtered_df.loc[mask_params]
403
+ print('step 3')
404
+ print(filtered_df['model_name_for_query'])
405
 
406
  return filtered_df
407