steve
/
factor
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

rosetta-code.probabilistic-choice: cleanup.

db4
John Benediktsson 2015-07-19 12:36:07 -07:00
parent c25fb5f55e
commit 109c7328bb
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
extra/rosetta-code/probabilistic-choice/probabilistic-choice.factor
View File

@ -1,7 +1,7 @@
! Copyright (c) 2012 Anonymous ! Copyright (c) 2012 Anonymous
! See http://factorcode.org/license.txt for BSD license. ! See http://factorcode.org/license.txt for BSD license.
USING: arrays assocs combinators.random io kernel macros math USING: arrays assocs combinators.random formatting fry kernel
math.statistics prettyprint quotations sequences sorting formatting ; macros math quotations sequences ;
IN: rosettacode.probabilistic-choice IN: rosettacode.probabilistic-choice
! http://rosettacode.org/wiki/Probabilistic_choice ! http://rosettacode.org/wiki/Probabilistic_choice
@ -38,25 +38,25 @@ CONSTANT: data
} }
MACRO: case-probas ( data -- quot ) MACRO: case-probas ( data -- quot )
[ first2 [ swap 1quotation 2array ] [ 1quotation ] if* ] map 1quotation ; [ first2 [ 1quotation ] dip [ swap 2array ] when* ] map 1quotation ;
: expected ( name data -- float ) : expected ( data name -- float )
2dup at [ 2nip ] [ nip values sift sum 1 swap - ] if* ; dupd of [ nip ] [ values sift sum 1 swap - ] if* ;
: generate ( # case-probas -- seq ) : generate ( # case-probas -- seq )
H{ } clone H{ } clone [
[ [ [ casep ] [ inc-at ] bi* ] 2curry times ] keep ; inline '[ _ casep _ inc-at ] times
] keep ; inline
: normalize ( seq # -- seq ) : normalize ( seq # -- seq )
[ clone ] dip [ /f ] curry assoc-map ; [ clone ] dip '[ _ /f ] assoc-map ;
: summarize1 ( name value data -- ) : summarize1 ( name value data -- )
[ over ] dip expected pick expected "%6s: %10f %10f\n" printf ;
"%6s: %10f %10f\n" printf ;
: summarize ( generated data -- ) : summarize ( generated data -- )
"Key" "Value" "expected" "%6s %10s %10s\n" printf "Key" "Value" "expected" "%6s %10s %10s\n" printf
[ summarize1 ] curry assoc-each ; '[ _ summarize1 ] assoc-each ;
: generate-normalized ( # proba -- seq ) : generate-normalized ( # proba -- seq )
[ generate ] [ drop normalize ] 2bi ; inline [ generate ] [ drop normalize ] 2bi ; inline