]> gitweb.factorcode.org Git - factor.git/blobdiff - basis/math/statistics/statistics.factor
core: Add the shuffler words but without primitives.
[factor.git] / basis / math / statistics / statistics.factor
index 79bbeb046fbcb0acc3ff9b4b1a657c150fe1c203..32e0ce184ed0ceeb75d19afd9097d666826b269c 100644 (file)
@@ -269,7 +269,7 @@ ALIAS: std sample-std
 <PRIVATE
 : r-sum-diffs ( x-mean y-mean x-seq y-seq -- (r) )
     ! finds sigma((xi-mean(x))(yi-mean(y))
-    0 [ [ [ pick ] dip swap - ] bi@ * + ] 2reduce 2nip ;
+    0 [ [ reach - ] bi@ * + ] 2reduce 2nip ;
 
 : (r) ( x-mean y-mean x-seq y-seq x-std y-std -- r )
     * recip [ [ r-sum-diffs ] keep length 1 - / ] dip * ;
@@ -281,7 +281,7 @@ PRIVATE>
 : pearson-r ( xy-pairs -- r ) r-stats (r) ;
 
 : least-squares ( xy-pairs -- alpha beta )
-    r-stats [ 2dup ] 4 ndip
+    r-stats [ 2dup ] 4dip
     ! stack is x-mean y-mean x-mean y-mean x-seq y-seq x-std y-std
     [ (r) ] 2keep ! stack is mean(x) mean(y) r sx sy
     swap / * ! stack is mean(x) mean(y) beta