| 646 | return network_weights |
| 647 | |
| 648 | class MyDataset_trans(TensorDataset): |
| 649 | def __init__(self, trainval_df): |
| 650 | super(MyDataset_trans, self).__init__() |
| 651 | self.df = trainval_df |
| 652 | #self.df.reset_index(drop=True, inplace=True) |
| 653 | def __len__(self): |
| 654 | return len(self.df) |
| 655 | |
| 656 | def __getitem__(self, index): |
| 657 | |
| 658 | return (self.df.loc[index,0], self.df.loc[index,1], self.df.loc[index,2], self.df.loc[index,3], \ |
| 659 | self.df.loc[index,4], self.df.loc[index,5]) |
| 660 | |
| 661 | |
| 662 | def k_fold_trainer_graph_trans(temp_loader_trainval,model,args): |
nothing calls this directly
no outgoing calls
no test coverage detected