(symbols: np.array, bitwidth: int=8)
| 280 | return params |
| 281 | |
| 282 | def _get_prob(symbols: np.array, bitwidth: int=8) -> np.array: |
| 283 | prob_list = [] |
| 284 | for i in range(symbols.shape[-1]): |
| 285 | counts = np.bincount(symbols[..., i].ravel(), minlength=2**bitwidth) |
| 286 | |
| 287 | prob = counts / counts.sum() |
| 288 | prob = prob.astype(np.float32) |
| 289 | |
| 290 | prob_list.append(prob) |
| 291 | |
| 292 | stacked_prob_npy = np.stack(prob_list, axis=0) |
| 293 | |
| 294 | return stacked_prob_npy |
| 295 | |
| 296 | def _get_likelihood(symbols: np.array, bitwidth: int=8) -> np.array: |
| 297 | pass |
no outgoing calls
no test coverage detected