| 868 | return network_weights |
| 869 | |
| 870 | class MyDataset_pairwise(TensorDataset): |
| 871 | def __init__(self, trainval_df): |
| 872 | super(MyDataset_pairwise, self).__init__() |
| 873 | self.df = trainval_df |
| 874 | #self.df.reset_index(drop=True, inplace=True) |
| 875 | def __len__(self): |
| 876 | return len(self.df) |
| 877 | |
| 878 | def __getitem__(self, index): |
| 879 | |
| 880 | return (self.df.loc[index,0], self.df.loc[index,1], self.df.loc[index,2], self.df.loc[index,3], self.df.loc[index,4]) |
| 881 | |
| 882 | |
| 883 | def k_fold_trainer_graph_pairwise(temp_loader_trainval,model,args): |
nothing calls this directly
no outgoing calls
no test coverage detected