Creates a KafkaDataset.
kafka_dataset( topics, servers = "localhost", group = "", eof = FALSE, timeout = 1000 )
| topics | A |
|---|---|
| servers | A list of bootstrap servers. |
| group | The consumer group id. |
| eof | If True, the kafka reader will stop on EOF. |
| timeout | The timeout value for the Kafka Consumer to wait (in millisecond). |
if (FALSE) { dataset <- kafka_dataset( topics = list("test:0:0:4"), group = "test", eof = TRUE) %>% dataset_repeat(1) sess <- tf$Session() iterator <- make_iterator_one_shot(dataset) next_batch <- iterator_get_next(iterator) until_out_of_range({ batch <- sess$run(next_batch) print(batch) }) }