我试图训练我的CNN模型,但我遇到了错误: AttributeError:模块“h5py”没有属性“Dataset”
train = train_data[:-500]
test = train_data[-500:]
X = np.array([i[0] for i in train]).reshape(-1,IMG_SIZE,IMG_SIZE,1)
Y = [i[1] for i in train]
test_x = np.array([i[0] for i in test]).reshape(-1,IMG_SIZE,IMG_SIZE,1)
test_y = [i[1] for i in test]
model.fit(X ,Y , n_epoch=3, validation_set=(test_x, test_y),
snapshot_step=500, show_metric=True, run_id=MODEL_NAME)
当我运行model.fit时,下面是我收到的错误
Exception in thread Thread-9:
Traceback (most recent call last):
File "C:\Users\nikhi\Anaconda3\envs\Dogs_vs_Cats\lib\threading.py", line 916, in _bootstrap_inner
self.run()
File "C:\Users\nikhi\Anaconda3\envs\Dogs_vs_Cats\lib\threading.py", line 864, in run
self._target(*self._args, **self._kwargs)
File "C:\Users\nikhi\Anaconda3\envs\Dogs_vs_Cats\lib\site-packages\tflearn\data_flow.py", line 187, in fill_feed_dict_queue
data = self.retrieve_data(batch_ids)
File "C:\Users\nikhi\Anaconda3\envs\Dogs_vs_Cats\lib\site-packages\tflearn\data_flow.py", line 222, in retrieve_data
utils.slice_array(self.feed_dict[key], batch_ids)
File "C:\Users\nikhi\Anaconda3\envs\Dogs_vs_Cats\lib\site-packages\tflearn\utils.py", line 184, in slice_array
if type(X) == h5py.Dataset:
AttributeError: module 'h5py' has no attribute 'Dataset'
目前没有回答
相关问题 更多 >
编程相关推荐