rowan.a.tl/propeller
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update downsample.cljc

Browse Source
This commit is contained in:
Ryan Boldi 2023-10-15 22:54:22 -04:00
parent e7130da06f
commit 288faadedf
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/propeller/downsample.cljc
View File

@ -6,14 +6,14 @@
(defn assign-indices-to-data (defn assign-indices-to-data
"assigns an index to each training case in order to differentiate them when downsampling" "assigns an index to each training case in order to differentiate them when downsampling"
[training-data] [training-data]
(map (fn [data-map index] (pmap (fn [data-map index]
(let [data-m (if (map? data-map) data-map (assoc {} :data data-map))] ;if data is not in a map, make it one (let [data-m (if (map? data-map) data-map (assoc {} :data data-map))] ;if data is not in a map, make it one
(assoc data-m :index index))) (assoc data-m :index index)))
training-data (range (count training-data)))) training-data (range (count training-data))))
(defn initialize-case-distances (defn initialize-case-distances
[{:keys [training-data population-size]}] [{:keys [training-data population-size] :as argmap}]
(map #(assoc % :distances (vec (repeat (count training-data) population-size))) training-data)) (utils/pmapallv #(assoc % :distances (vec (repeat (count training-data) population-size))) training-data argmap))
(defn select-downsample-random (defn select-downsample-random
"Selects a downsample from the training cases and returns it" "Selects a downsample from the training cases and returns it"