Heimdall.fe.ScBERTBinningFe.binning

Heimdall.fe.ScBERTBinningFe.binning#

ScBERTBinningFe.binning(row, n_bins)[source]#

scBERT-style binning: bin values into [0, n_bins], capping high values. - Zeros are retained as 0. - Values > n_bins are capped at n_bins.

Return type:

ndarray | Tensor