machine-learning: Add stratified-samples for getting multiple samples.

diff --git a/extra/machine-learning/rebalancing/rebalancing.factor b/extra/machine-learning/rebalancing/rebalancing.factor
index d95029f19706ef80e3e935a7eaaa63f8b17921ee..41c3296eee853e89b40e68fcda73842ecff99ef6 100644 (file)
--- 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