(self, cache_dir="data/SOTU/SOTU_QA.csv", sample_size=None)
| 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) |
| 90 | if sample_size is not None: |
| 91 | sampled_df = df.sample(sample_size, random_state=self.seed)[["question", "answer"]].reset_index(drop=True) |
| 92 | return sampled_df |
| 93 | return df |