(self, cache_dir="data/bigbench/physics_question.csv", sample_size=None)
| 58 | return sampled_df |
| 59 | |
| 60 | def load_physics_question(self, cache_dir="data/bigbench/physics_question.csv", sample_size=None): |
| 61 | df = pd.read_csv(cache_dir) |
| 62 | if sample_size is not None: |
| 63 | sampled_df = df.sample(sample_size, random_state=self.seed)[["input", "target"]].reset_index(drop=True) |
| 64 | return sampled_df |
| 65 | return df |
| 66 | |
| 67 | def load_sports_understanding(self, cache_dir="data/bigbench/sports_understanding.csv", sample_size=None): |
| 68 | df = pd.read_csv(cache_dir) |