Module Caisar_xgboost.TreeSource

Sourcetype tree =
  1. | Split of {
    1. split_indice : int;
    2. split_condition : float;
    3. left : tree;
    4. right : tree;
    5. missing : [ `Left ];
    }
  2. | Leaf of {
    1. leaf_value : float;
    }
Sourcetype op =
  1. | Identity
  2. | Sigmoid
Sourcetype t = {
  1. base_score : float;
  2. trees : tree array;
  3. after_sum : op;
}

the value is op(base_score + sum(tree))

Sourceval convert : Parser.t -> t
Sourceval predict : t -> Input.t -> float