init research
This commit is contained in:
@@ -0,0 +1,37 @@
|
||||
(ns tech.v3.dataset.metamorph-test
|
||||
(:require [tech.v3.dataset.metamorph :as ds-mm]
|
||||
[tech.v3.dataset :as ds]
|
||||
[clojure.test :as t :refer [deftest is]]))
|
||||
|
||||
(def df
|
||||
(ds/->dataset "test/data/ames-train.csv.gz" {:key-fn keyword}))
|
||||
|
||||
|
||||
(deftest call-with-df-1
|
||||
(is (= [1 2 3 4 5]
|
||||
(->>
|
||||
((ds-mm/set-inference-target :SalePrice) df)
|
||||
:metamorph/data
|
||||
:Id
|
||||
(take 5)
|
||||
))))
|
||||
|
||||
(deftest call-with-df-2
|
||||
(is (= [1 2 3 4 5]
|
||||
(->>
|
||||
((ds-mm/rename-columns {:SalePrice :sale-price :Id :id})
|
||||
{:metamorph/data df})
|
||||
((ds-mm/set-inference-target :sale-price) )
|
||||
:metamorph/data
|
||||
:id
|
||||
(take 5)))))
|
||||
|
||||
(deftest brief
|
||||
(let [df (ds/select-columns df (sort (ds/column-names df)))]
|
||||
(is (= 334.0
|
||||
(->>
|
||||
((ds-mm/brief)
|
||||
{:metamorph/data df})
|
||||
:metamorph/data
|
||||
first
|
||||
:min)))))
|
||||
Reference in New Issue
Block a user