TransWikia.com

Tensorflow v1 Dataset API AttributeError with ndim

Data Science Asked by AutomaKen on December 29, 2020

I’d like to make pipeline for optimizing Gpu and Cpu.
Dataset

It’s about 10000 datapoint and 4 description variables for the regression problem.

df = pd.read_csv("dataset")
X_train, X_test, y_train, y_test = 
train_test_split(df.iloc[:, :-1].values, df.iloc[:, -1].values)
scaler = MinMaxScaler()
scaler.fit(X_train)
X_train_scaled = scaler.transform(X_train)
batch_size = 64
with tf.Session() as sess:
  dataset = tf.data.Dataset.from_tensor_slices((X_train_scaled, y_train))
  dataset = dataset.cache()
  dataset = dataset.shuffle(len(X_train_scaled))
  dataset = dataset.repeat()
  dataset = dataset.batch(batch_size)
  dataset = dataset.prefetch(batch_size*10)
  iterator = dataset.make_one_shot_iterator()
  print(sess.run(iterator.get_next()[0]),sess.run(iterator.get_next()[1]))
  his = model.fit(dataset, epochs=300, steps_per_epoch=1000, verbose=0)

[[0.54192635 0.36815166 0.37738184 0.13592493] [0.31898017 0.33687204
0.59490225 0.59597855] [0.2733711 0.26047393 0.42761693 0.99986595] [0.77025496 0.98919431 0.45632269 0.66447721] [0.64305949 0.50236967
0.53823311 0.56313673]]
[429.66 460.53 428.49 446.62 456.84]

In the fitting model, the following error

AttributeError: 'PrefetchDataset' object has no attribute 'ndim'

I saw some issues with this problem.

But it didn’t work for me.

Software version:
Keras:2.2.4
Tensorflow:1.12.0

Add your own answers!

Ask a Question

Get help from others!

© 2024 TransWikia.com. All rights reserved. Sites we Love: PCI Database, UKBizDB, Menu Kuliner, Sharing RPP