Create dataset file for each perturbation condition
(self)
| 1362 | return cell_graphs |
| 1363 | |
| 1364 | def create_dataset_file(self): |
| 1365 | """ |
| 1366 | Create dataset file for each perturbation condition |
| 1367 | """ |
| 1368 | print_sys("Creating dataset file...") |
| 1369 | self.dataset_processed = {} |
| 1370 | for p in tqdm(self.adata.obs['condition'].unique()): |
| 1371 | self.dataset_processed[p] = self.create_cell_graph_dataset(self.adata, p) |
| 1372 | print_sys("Done!") |
| 1373 | |
| 1374 | |
| 1375 | def main(data_path='./data', out_dir='./saved_models', device='cuda:0'): |
no test coverage detected