extra/machine-learning/label-binarizer/label-binarizer-tests.factor

   1 ! Copyright (C) 2012 Doug Coleman.
   2 ! See http://factorcode.org/license.txt for BSD license.
   3 USING: kernel machine-learning.label-binarizer
   4 machine-learning.transformer tools.test ;
   5
   6 { { { 1 0 0 0 } { 0 0 0 1 } } } [
   7     <label-binarizer> { 1 2 6 4 2 } over fit-y
   8     { 1 6 } swap transform-y
   9 ] unit-test
  10
  11 { { 1 6 } } [
  12     <label-binarizer> { 1 2 6 4 2 } over fit-y
  13     { 1 6 } over transform-y swap inverse-transform-y
  14 ] unit-test