debug
Browse filesSigned-off-by: n1ck-guo <heng.guo@intel.com>
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 |
|