init research

2026-02-08 11:20:43 -10:00
commit bdf064f54d
3041 changed files with 1592200 additions and 0 deletions
@@ -0,0 +1,26 @@
+(ns tech.v3.dataset.test-utils
+  (:require [tech.v3.io :as io]
+            [clojure.string :as s]
+            [tech.v3.dataset :as ds]
+            [camel-snake-kebab.core :refer [->kebab-case]]))
+
+(defn load-mapseq-fruit-dataset
+  []
+  (let [fruit-ds (slurp (io/input-stream "test/data/fruit_data_with_colors.txt"))
+        dataset (->> (s/split fruit-ds #"\n")
+                     (mapv #(s/split % #"\s+")))
+        ds-keys (->> (first dataset)
+                     (mapv (comp keyword ->kebab-case)))]
+    (->> (rest dataset)
+         (map (fn [ds-line]
+                (->> ds-line
+                     (map (fn [ds-val]
+                            (try
+                              (Double/parseDouble ^String ds-val)
+                              (catch Throwable e
+                                (-> (->kebab-case ds-val)
+                                    keyword)))))
+                     (zipmap ds-keys))))
+         (ds/->dataset))))
+
+(def mapseq-fruit-dataset (memoize load-mapseq-fruit-dataset))