Skip to content

Instantly share code, notes, and snippets.

@johnptoohey
Forked from jackrusher/SimpleMarkovChain.clj
Created July 23, 2013 15:51
Show Gist options
  • Select an option

  • Save johnptoohey/6063507 to your computer and use it in GitHub Desktop.

Select an option

Save johnptoohey/6063507 to your computer and use it in GitHub Desktop.
(defn model-from-sequence
"Returns a transition matrix of 'depth' from 'sequence'"
[depth sequence]
(loop [accum {} chunks (partition (inc depth) 1 (seq sequence))]
(if (seq? chunks)
(let [chunk (first chunks)
prefix (drop-last chunk)
suffix (last chunk)]
(recur (assoc accum prefix (conj (get accum prefix []) suffix)) (next chunks)))
accum)))
(defn chain
"Returns a lazy Markov chain starting with 'current' using matrix 'trans'"
[current trans]
(if-let [transitions (trans current)]
(cons (first current)
(lazy-seq (chain (concat (rest current) [(rand-nth transitions)]) trans)))
current))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment