diff --git a/extra/machine-learning/rebalancing/rebalancing.factor b/extra/machine-learning/rebalancing/rebalancing.factor index d95029f197..41c3296eee 100644 --- a/extra/machine-learning/rebalancing/rebalancing.factor +++ b/extra/machine-learning/rebalancing/rebalancing.factor @@ -22,6 +22,9 @@ MEMO: probabilities-seq ( seq -- seq' ) : stratified-sample ( stratified-sequences probability-sequence -- elt ) probabilities-quot call swap nth random ; inline +: stratified-samples ( stratified-sequences probability-sequence n -- elt ) + [ '[ _ _ stratified-sample ] ] dip swap replicate ; + : equal-stratified-sample ( stratified-sequences -- elt ) random random ; inline