Fixing WeightedSamplingReader that got stale

uber / petastorm

Petastorm library enables single machine or distributed training and evaluation of deep learning models from datasets in Apache Parquet format. It supports ML frameworks such as Tensorflow, Pytorch, and PySpark and can be used from pure Python code.

Apache License 2.0

1.8k stars 284 forks source link

Fixing WeightedSamplingReader that got stale #434

Closed selitvin closed 5 years ago

selitvin commented 5 years ago

Adding some tests to cover WeightedSamplingReader usage from TF as well as some newly added argument validation.