(self, cache_dir="data/bigbench/strategy_qa.csv", sample_size=None)
| 79 | return df |
| 80 | |
| 81 | def load_strategy_qa(self, cache_dir="data/bigbench/strategy_qa.csv", sample_size=None): |
| 82 | df = pd.read_csv(cache_dir) |
| 83 | if sample_size is not None: |
| 84 | sampled_df = df.sample(sample_size, random_state=self.seed)[["input", "target"]].reset_index(drop=True) |
| 85 | return sampled_df |
| 86 | return df |
| 87 | |
| 88 | def load_sotu_qa(self, cache_dir="data/SOTU/SOTU_QA.csv", sample_size=None): |
| 89 | df = pd.read_csv(cache_dir) |