MCPcopy Create free account
hub / github.com/dmlc/parameter_server / read

Method read

src/learner/sgd.h:131–150  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

129 }
130
131 bool read(MatrixPtr<V>& Y, MatrixPtr<V>& X, SArray<Key>& key) {
132 MatrixPtrList<V> data;
133 if (!data_prefetcher_.pop(&data)) return false;
134 CHECK_EQ(data.size(), 2);
135 Y = data[0];
136
137 // localizer
138 SArray<Key> uniq_key;
139 SArray<uint8> key_cnt;
140 Localizer<Key, V> localizer;
141 localizer.countUniqIndex(data[1], &uniq_key, &key_cnt);
142
143 // filter keys
144 filter_.insertKeys(uniq_key, key_cnt);
145 key = filter_.queryKeys(uniq_key, key_freq_);
146
147 // to local keys
148 X = localizer.remapIndex(key);
149 return true;
150 }
151 private:
152 int minibatch_size_ = 1000;
153 StreamReader<V> reader_;

Callers 1

loadDataMethod · 0.45

Calls 6

insertKeysMethod · 0.80
queryKeysMethod · 0.80
remapIndexMethod · 0.80
popMethod · 0.45
sizeMethod · 0.45
countUniqIndexMethod · 0.45

Tested by

no test coverage detected