Module Smtzilla_utils.Regression_modelSource

Sourcemodule FeatMap : sig ... end
Sourcetype features = int FeatMap.t
Sourcetype score
Sourceval compare_score : score -> score -> int
Sourceval score_of_float : float -> score
Sourceval pp_score : score Fmt.t
Sourcetype tree =
  1. | Leaf of score
  2. | Node of {
    1. feature : string;
    2. threshold : score;
    3. left : tree;
    4. right : tree;
    }
Sourcetype gb_model = {
  1. init_value : score;
  2. trees : tree list;
}
Sourcetype dt_model = tree
Sourcetype t =
  1. | GBModel of gb_model
  2. | DTModel of dt_model
Sourceval pp : t Fmt.t
Sourceval read_models_from_file : string -> (string * t) list
Sourceval choose_best : (score * 'a) list -> 'a
Sourceval predict : features -> t -> score