Searched refs:reshape_function (Results 1 – 2 of 2) sorted by relevance
36 def reshape_function(data, label): function129 train_data = train_data.map(reshape_function)130 test_data = test_data.map(reshape_function)131 valid_data = valid_data.map(reshape_function)
30 from train import reshape_function66 self.train_data = self.train_data.map(reshape_function)
