Module Smtzilla_utils.Regression_model

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