HaskellGBM-0.1.0.0

Safe HaskellNone
LanguageHaskell2010

LightGBM.Model

Contents

Synopsis

Models

data Model Source #

A model to use to make predictions

Instances

Eq Model Source # 

Methods

(==) :: Model -> Model -> Bool #

(/=) :: Model -> Model -> Bool #

Show Model Source # 

Methods

showsPrec :: Int -> Model -> ShowS #

show :: Model -> String #

showList :: [Model] -> ShowS #

trainNewModel Source #

Arguments

:: [Param]

Training parameters

-> DataSet

Training data

-> [DataSet]

Testing data

-> IO (Either ErrLog Model) 

Train a new model and persist it to a file.

readModelFile :: FilePath -> IO Model Source #

Persisted models can be loaded up and used for prediction.

writeModelFile :: FilePath -> Model -> IO () Source #

Models can be written out to a file

Prediction

predict Source #

Arguments

:: Model

A model to do prediction with

-> [Param]

Parameters

-> [PredictionParam] 
-> DataSet

The new input data for prediction

-> IO (Either ErrLog DataSet)

The prediction output DataSet

Predict the results of new inputs and persist the results to an output file.