package modelkit

  1. Overview
  2. Docs
package modelkit
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Module type Modelkit.ESTIMATORSource

Common contract for immutable estimator specifications.

t is a training specification and fitted is the value produced by fitting it. Implementations receive RNG state explicitly.

include SPECIFICATION
Sourcetype t
Sourcetype params
Sourceval clone : t -> t
Sourceval params : t -> params
Sourcetype target
Sourcetype prediction
Sourcetype fitted
Sourcetype rng
Sourceval fit : t -> ?sample_weight:Sample_weight.t -> rng:rng -> feature_schema:Feature_schema.t -> x:Matrix.t -> y:target -> unit -> (fitted, Error.t) result
Sourceval predict : fitted -> feature_schema:Feature_schema.t -> x:Matrix.t -> (prediction, Error.t) result
Sourceval fitted_params : fitted -> params
Sourceval feature_schema : fitted -> Feature_schema.t