forked from zero-one-group/geni
-
Notifications
You must be signed in to change notification settings - Fork 0
/
fpm.clj
23 lines (21 loc) · 806 Bytes
/
fpm.clj
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
(ns zero-one.geni.ml.fpm
(:require
[zero-one.geni.interop :as interop])
(:import
(org.apache.spark.ml.fpm FPGrowth
PrefixSpan)))
(defn fp-growth [params]
(let [defaults {:items-col "items",
:min-confidence 0.8,
:min-support 0.3,
:prediction-col "prediction"}
props (merge defaults params)]
(interop/instantiate FPGrowth props)))
(def frequent-pattern-growth fp-growth)
(defn prefix-span [params]
(let [defaults {:min-support 0.1,
:sequence-col "sequence",
:max-pattern-length 10,
:max-local-proj-db-size 32000000}
props (merge defaults params)]
(interop/instantiate PrefixSpan props)))