急急急 各位大佬帮我看看map函数

我上岸啦

我想请问一下在TensorFlow的dataset模块中有个map函数功能，我看论坛上都是直接调用编写好的函数，可是我在map之前已经写好了一个函数，为什么会提示没有函数的呢？

以下是我的代码，运行后报错了这个
TypeError: map() missing 1 required positional argument: 'map_func'

1. import tensorflow as tf
   
2. import numpy as np
   
3. import os
   
4. import gzip
   
5. 
   
6. 
   
7. def load_data(data_folder):
   
8. 
   
9.   files = [
   
10.       'train-labels-idx1-ubyte.gz', 'train-images-idx3-ubyte.gz',
    
11.       't10k-labels-idx1-ubyte.gz', 't10k-images-idx3-ubyte.gz'
    
12.   ]
    
13. 
    
14.   paths = []
    
15.   for fname in files:
    
16.     paths.append(os.path.join(data_folder,fname))
    
17. 
    
18.   with gzip.open(paths[0], 'rb') as lbpath:
    
19.     y_train = np.frombuffer(lbpath.read(), np.uint8, offset=8)
    
20. 
    
21.   with gzip.open(paths[1], 'rb') as imgpath:
    
22.     x_train = np.frombuffer(
    
23.         imgpath.read(), np.uint8, offset=16).reshape(len(y_train), 28, 28)
    
24. 
    
25.   with gzip.open(paths[2], 'rb') as lbpath:
    
26.     y_test = np.frombuffer(lbpath.read(), np.uint8, offset=8)
    
27. 
    
28.   with gzip.open(paths[3], 'rb') as imgpath:
    
29.     x_test = np.frombuffer(
    
30.         imgpath.read(), np.uint8, offset=16).reshape(len(y_test), 28, 28)
    
31. 
    
32.   return (x_train, y_train), (x_test, y_test)
    
33. 
    
34. (train_X,trian_label),(test_X,test_label) = load_data('C:\\Users\\Administrator\\Desktop\\data')
    
35. 
    
36. 
    
37. def func(x,y):
    
38.         x = np.expand_dims(x.astype(np.float32) / 255.0,axis = -1)
    
39.         return x,y
    
40. 
    
41. dataset = tf.data.Dataset.map(func)

复制代码

zltzlt

1. import tensorflow as tf
   
2. import numpy as np
   
3. import os
   
4. import gzip
   
5. 
   
6. 
   
7. def load_data(data_folder):
   
8.     files = [
   
9.         'train-labels-idx1-ubyte.gz', 'train-images-idx3-ubyte.gz',
   
10.         't10k-labels-idx1-ubyte.gz', 't10k-images-idx3-ubyte.gz'
    
11.     ]
    
12. 
    
13.     paths = []
    
14.     for fname in files:
    
15.         paths.append(os.path.join(data_folder, fname))
    
16. 
    
17.     with gzip.open(paths[0], 'rb') as lbpath:
    
18.         y_train = np.frombuffer(lbpath.read(), np.uint8, offset=8)
    
19. 
    
20.     with gzip.open(paths[1], 'rb') as imgpath:
    
21.         x_train = np.frombuffer(
    
22.             imgpath.read(), np.uint8, offset=16).reshape(len(y_train), 28, 28)
    
23. 
    
24.     with gzip.open(paths[2], 'rb') as lbpath:
    
25.         y_test = np.frombuffer(lbpath.read(), np.uint8, offset=8)
    
26. 
    
27.     with gzip.open(paths[3], 'rb') as imgpath:
    
28.         x_test = np.frombuffer(
    
29.             imgpath.read(), np.uint8, offset=16).reshape(len(y_test), 28, 28)
    
30. 
    
31.     return (x_train, y_train), (x_test, y_test)
    
32. 
    
33. 
    
34. (train_X, trian_label), (test_X, test_label) = load_data('C:\\Users\\Administrator\\Desktop\\data')
    
35. 
    
36. 
    
37. def func(x, y):
    
38.     x = np.expand_dims(x.astype(np.float32) / 255.0, axis=-1)
    
39.     return x, y
    
40. 
    
41. 
    
42. dataset = tf.data.Dataset().map(func)

复制代码

我上岸啦

zltzlt 发表于 2020-8-15 15:42

大佬好 可是还是报错呀
Can't instantiate abstract class DatasetV2 with abstract methods _inputs, element_spec
但是这次是报错这个