Searched refs:data_folder (Results 1 – 2 of 2) sorted by relevance
166 def get_train_val_data(data_folder: Path, regen_data: bool, train_percentage=0.95):169 and Path(data_folder / "train_df.csv").exists()170 and Path(data_folder / "val_df.csv").exists()172 train_data = pd.read_csv(data_folder / "train_df.csv")173 val_data = pd.read_csv(data_folder / "val_df.csv")177 commit_list_df = pd.read_csv(data_folder / "commitlist.csv")192 test_df.to_csv(data_folder / "test_df.csv", index=False)193 train_df.to_csv(data_folder / "train_df.csv", index=False)194 val_df.to_csv(data_folder / "val_df.csv", index=False)198 def get_author_map(data_folder: Path, regen_data, assert_stored=False):[all …]
68 def __init__(self, data_folder): argument69 self.data_folder = data_folder73 filename = os.path.join(self.data_folder,