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