-- Hoogle documentation, generated by Haddock
-- See Hoogle, http://www.haskell.org/hoogle/


-- | Generic encoding of records
--   
--   Generic encoding of records. It currently provides a single,
--   polymorphic function to encode sum types (i.e. categorical variables)
--   as one-hot vectors.
@package record-encode
@version 0.2.1


-- | This library provides generic machinery (via GHC.Generics and
--   `generics-sop`) to encode values of some algebraic type as points in a
--   vector space.
--   
--   Processing datasets that have one or more categorical variables (which
--   in other words are values of a sum type) typically requires a series
--   of boilerplate transformations, and the <a>encodeOneHot</a> function
--   provided here does precisely that.
--   
--   <h2>Internals</h2>
--   
--   This library makes use of generic programming to analyze both values
--   and types (see the internal Data.Record.Encode.Generics module).
--   
--   Initially, it was relying on Template Haskell to analyze <i>types</i>,
--   using the the instance generation machinery explained here:
--   <a>https://markkarpov.com/tutorial/th.html#example-1-instance-generation</a>
module Data.Record.Encode

-- | Computes the one-hot encoding of a value of a sum type.
--   
--   The type of the input value must be an instance of <a>Generic</a>
--   (from GHC.Generics) <i>and</i> of <a>Generic</a> (from the
--   `generics-sop` library).
--   
--   <pre>
--   &gt;&gt;&gt; :set -XDeriveGeneric
--   </pre>
--   
--   <pre>
--   &gt;&gt;&gt; import qualified GHC.Generics as G
--   
--   &gt;&gt;&gt; import qualified Generics.SOP as SOP
--   
--   &gt;&gt;&gt; import Data.Record.Encode
--   </pre>
--   
--   <pre>
--   &gt;&gt;&gt; data X = A | B | C deriving (G.Generic)
--   
--   &gt;&gt;&gt; instance SOP.Generic X
--   </pre>
--   
--   <pre>
--   &gt;&gt;&gt; encodeOneHot B
--   OH {oDim = 3, oIx = 1}
--   </pre>
encodeOneHot :: forall a. G a => a -> OneHot

-- | A one-hot encoding is a d-dimensional vector having a single component
--   equal to 1 and all others equal to 0.
data OneHot
OH :: !Int -> !Int -> OneHot

-- | Dimension of ambient space (i.e. number of categories)
[oDim] :: OneHot -> !Int

-- | Index of nonzero entry
[oIx] :: OneHot -> !Int
compareOH :: OneHot -> OneHot -> Maybe Ordering

-- | Create a one-hot vector
oneHotV :: Num a => OneHot -> Vector a

-- | Constraints necessary to <a>encodeOneHot</a> a value.
--   
--   NB: <a>GVariants</a> is an internal typeclass, and this constraint is
--   automatically satisfied if the type is an instance of <a>Generic</a>
type G a = (GVariants (Rep a), Generic a, Generic a)
instance GHC.Show.Show Data.Record.Encode.OneHot
instance GHC.Classes.Eq Data.Record.Encode.OneHot
instance GHC.Generics.Generic Data.Record.Encode.X
instance Generics.SOP.Universe.Generic Data.Record.Encode.X