Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def pandas_input_fn(x, y=None, batch_size=128, num_epochs=None):
- def input_fn():
- if y is not None:
- x.loc[:,'y'] = y
- queue = learn.dataframe.queues.feeding_functions.enqueue_data(
- x, 1000, shuffle=num_epochs is None, num_epochs=num_epochs)
- if num_epochs is None:
- features = queue.dequeue_many(batch_size)
- else:
- features = queue.dequeue_up_to(batch_size)
- features = dict(zip(['index'] + list(x.columns), features))
- if y is not None:
- target = features.pop('y')
- return features, target
- return features
- return input_fn
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement