Searched refs:cat_feature2cats (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torchgen/_autoheuristic/ |
| H A D | train.py | 107 def add_real_datasets(self, datasets, other_datasets, cat_feature2cats): argument
111 path, cat_feature2cats=cat_feature2cats, apply_filters=False
116 self, cat_feature2cats, categorical_features, results argument
121 if not cat_feature2cats:
122 cat_feature2cats = {}
124 if cat_feature in cat_feature2cats:
125 categories = cat_feature2cats[cat_feature]
128 cat_feature2cats[cat_feature] = categories
140 return (results, cat_feature2cats, dummy_col_2_col_val)
|
| H A D | train_decision.py | 250 def prepare_datasets(self, df, other_datasets, cat_feature2cats, ranking=False): argument
267 self.add_real_datasets(datasets, other_datasets, cat_feature2cats, ranking)
312 (df, choices, cat_feature2cats, dummy_col_2_col_val, metadata) = self.get_df(
316 datasets = self.prepare_datasets(df, other_datasets, cat_feature2cats, ranking)
383 cat_feature2cats=None, argument
500 cat_feature2cats,
503 cat_feature2cats, categorical_features, results
505 return (results, choices, cat_feature2cats, dummy_col_2_col_val, metadata)
584 self, datasets, other_datasets, cat_feature2cats, ranking=False argument
593 cat_feature2cats=cat_feature2cats,
|
| H A D | train_regression.py | 49 (df, choices, cat_feature2cats, dummy_col_2_col_val, metadata) = self.get_df(
54 self.add_real_datasets(datasets, other_datasets, cat_feature2cats)
85 def get_df(self, log_path, cat_feature2cats=None, nrows=None, apply_filters=False): argument
164 cat_feature2cats,
167 cat_feature2cats, categorical_features, results
169 return (results, choices, cat_feature2cats, dummy_col_2_col_val, metadata)
|