| 107 | ~MinibatchReader() { } |
| 108 | |
| 109 | void setReader(const DataConfig& file, int minibatch_size, int data_buf = 1000) { |
| 110 | reader_.init(file); |
| 111 | minibatch_size_ = minibatch_size; |
| 112 | data_prefetcher_.setCapacity(data_buf); |
| 113 | } |
| 114 | |
| 115 | void setFilter(size_t n, int k, int freq) { |
| 116 | filter_.resize(n, k); |
no test coverage detected