| 145 | } |
| 146 | |
| 147 | fn sample_embedding(&self, embedding: &Self::IndexingInput) { |
| 148 | for pair in &embedding.1 { |
| 149 | let value = pair.1; |
| 150 | |
| 151 | if value > 1.0 { |
| 152 | self.sampling_data.above_1.fetch_add(1, Ordering::Relaxed); |
| 153 | } |
| 154 | |
| 155 | if value > 2.0 { |
| 156 | self.sampling_data.above_2.fetch_add(1, Ordering::Relaxed); |
| 157 | } |
| 158 | |
| 159 | if value > 3.0 { |
| 160 | self.sampling_data.above_3.fetch_add(1, Ordering::Relaxed); |
| 161 | } |
| 162 | |
| 163 | if value > 4.0 { |
| 164 | self.sampling_data.above_4.fetch_add(1, Ordering::Relaxed); |
| 165 | } |
| 166 | |
| 167 | if value > 5.0 { |
| 168 | self.sampling_data.above_5.fetch_add(1, Ordering::Relaxed); |
| 169 | } |
| 170 | |
| 171 | if value > 6.0 { |
| 172 | self.sampling_data.above_6.fetch_add(1, Ordering::Relaxed); |
| 173 | } |
| 174 | |
| 175 | if value > 7.0 { |
| 176 | self.sampling_data.above_7.fetch_add(1, Ordering::Relaxed); |
| 177 | } |
| 178 | |
| 179 | if value > 8.0 { |
| 180 | self.sampling_data.above_8.fetch_add(1, Ordering::Relaxed); |
| 181 | } |
| 182 | |
| 183 | if value > 9.0 { |
| 184 | self.sampling_data.above_9.fetch_add(1, Ordering::Relaxed); |
| 185 | } |
| 186 | |
| 187 | self.sampling_data |
| 188 | .values_collected |
| 189 | .fetch_add(1, Ordering::Relaxed); |
| 190 | } |
| 191 | } |
| 192 | |
| 193 | fn finalize_sampling( |
| 194 | &self, |