Advertisement
Guest User

Untitled

a guest
Feb 22nd, 2019
76
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.78 KB | None | 0 0
  1. df = pd.read_csv("dataset")
  2. X_train, X_test, y_train, y_test =
  3. train_test_split(df.iloc[:, :-1].values, df.iloc[:, -1].values)
  4. scaler = MinMaxScaler()
  5. scaler.fit(X_train)
  6. X_train_scaled = scaler.transform(X_train)
  7. batch_size = 64
  8. with tf.Session() as sess:
  9. dataset = tf.data.Dataset.from_tensor_slices((X_train_scaled, y_train))
  10. dataset = dataset.cache()
  11. dataset = dataset.shuffle(len(X_train_scaled))
  12. dataset = dataset.repeat()
  13. dataset = dataset.batch(batch_size)
  14. dataset = dataset.prefetch(batch_size*10)
  15. iterator = dataset.make_one_shot_iterator()
  16. print(sess.run(iterator.get_next()[0]),sess.run(iterator.get_next()[1]))
  17. his = model.fit(dataset, epochs=300, steps_per_epoch=1000, verbose=0)
  18.  
  19. AttributeError: 'PrefetchDataset' object has no attribute 'ndim'
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement