caisar-xgboost 0.2 (latest)

From https://github.com/dmlc/xgboost/raw/master/doc/model.schema

val assoc : 
  (string
   * (Yojson.Safe.t ->
   (string * Yojson.Safe.t) list ->
   'a Ppx_deriving_yojson_runtime.error_or))
    list ->
  Yojson.Safe.t ->
  'a Ppx_deriving_yojson_runtime.error_or

type int_option = int option

val pp_int_option : 
  Ppx_deriving_runtime.Format.formatter ->
  int_option ->
  Ppx_deriving_runtime.unit

val show_int_option : int_option -> Ppx_deriving_runtime.string

val int_option_none : int32

val int_option_of_yojson : Yojson.Safe.t -> (int option, string) result

val int_option_to_yojson : int option -> [> `Intlit of string ]

type gbtree_model_param = {

num_trees : string;
num_parallel_tree : string;
size_leaf_vector : string;

}

and tree_param = {

num_nodes : string;
size_leaf_vector : string;
num_feature : string;

}

and reg_loss_param = {

scale_pos_weight : string;

}

and pseudo_huber_param = {

huber_slope : string;

}

and aft_loss_param = {

aft_loss_distribution : string;
aft_loss_distribution_scale : string;

}

and softmax_multiclass_param = {

num_class : string;

}

and lambda_rank_param = {

num_pairsample : string;
fix_list_weight : string;

}

and tree = {

tree_param : tree_param;
id : int;
loss_changes : float array;
sum_hessian : float array;
base_weights : float array;
left_children : int array;
right_children : int array;
parents : int_option array;
split_indices : int array;
split_conditions : float array;
split_type : int array;
default_left : bool array;
categories : int array;
categories_nodes : int array;
categories_segments : int array;
categories_sizes : int array;

}

and gbtree = {

gbtree_model_param : gbtree_model_param;
trees : tree array;
tree_info : int array;

}

and gblinear = {

weights : float array;

}

and dart = {

gbtree : gbtree;
weight_drop : float array;

}

and learner_model_param = {

base_score : string;
num_class : string;
num_feature : string;

}

val pp_gbtree_model_param : 
  Ppx_deriving_runtime.Format.formatter ->
  gbtree_model_param ->
  Ppx_deriving_runtime.unit

val show_gbtree_model_param : gbtree_model_param -> Ppx_deriving_runtime.string

val pp_tree_param : 
  Ppx_deriving_runtime.Format.formatter ->
  tree_param ->
  Ppx_deriving_runtime.unit

val show_tree_param : tree_param -> Ppx_deriving_runtime.string

val pp_reg_loss_param : 
  Ppx_deriving_runtime.Format.formatter ->
  reg_loss_param ->
  Ppx_deriving_runtime.unit

val show_reg_loss_param : reg_loss_param -> Ppx_deriving_runtime.string

val pp_pseudo_huber_param : 
  Ppx_deriving_runtime.Format.formatter ->
  pseudo_huber_param ->
  Ppx_deriving_runtime.unit

val show_pseudo_huber_param : pseudo_huber_param -> Ppx_deriving_runtime.string

val pp_aft_loss_param : 
  Ppx_deriving_runtime.Format.formatter ->
  aft_loss_param ->
  Ppx_deriving_runtime.unit

val show_aft_loss_param : aft_loss_param -> Ppx_deriving_runtime.string

val pp_softmax_multiclass_param : 
  Ppx_deriving_runtime.Format.formatter ->
  softmax_multiclass_param ->
  Ppx_deriving_runtime.unit

val show_softmax_multiclass_param : 
  softmax_multiclass_param ->
  Ppx_deriving_runtime.string

val pp_lambda_rank_param : 
  Ppx_deriving_runtime.Format.formatter ->
  lambda_rank_param ->
  Ppx_deriving_runtime.unit

val show_lambda_rank_param : lambda_rank_param -> Ppx_deriving_runtime.string

val pp_tree : 
  Ppx_deriving_runtime.Format.formatter ->
  tree ->
  Ppx_deriving_runtime.unit

val show_tree : tree -> Ppx_deriving_runtime.string

val pp_gbtree : 
  Ppx_deriving_runtime.Format.formatter ->
  gbtree ->
  Ppx_deriving_runtime.unit

val show_gbtree : gbtree -> Ppx_deriving_runtime.string

val pp_gblinear : 
  Ppx_deriving_runtime.Format.formatter ->
  gblinear ->
  Ppx_deriving_runtime.unit

val show_gblinear : gblinear -> Ppx_deriving_runtime.string

val pp_dart : 
  Ppx_deriving_runtime.Format.formatter ->
  dart ->
  Ppx_deriving_runtime.unit

val show_dart : dart -> Ppx_deriving_runtime.string

val pp_learner_model_param : 
  Ppx_deriving_runtime.Format.formatter ->
  learner_model_param ->
  Ppx_deriving_runtime.unit

val show_learner_model_param : 
  learner_model_param ->
  Ppx_deriving_runtime.string

val gbtree_model_param_to_yojson : gbtree_model_param -> Yojson.Safe.t

val gbtree_model_param_of_yojson : 
  Yojson.Safe.t ->
  gbtree_model_param Ppx_deriving_yojson_runtime.error_or

val tree_param_to_yojson : tree_param -> Yojson.Safe.t

val tree_param_of_yojson : 
  Yojson.Safe.t ->
  tree_param Ppx_deriving_yojson_runtime.error_or

val reg_loss_param_to_yojson : reg_loss_param -> Yojson.Safe.t

val reg_loss_param_of_yojson : 
  Yojson.Safe.t ->
  reg_loss_param Ppx_deriving_yojson_runtime.error_or

val pseudo_huber_param_to_yojson : pseudo_huber_param -> Yojson.Safe.t

val pseudo_huber_param_of_yojson : 
  Yojson.Safe.t ->
  pseudo_huber_param Ppx_deriving_yojson_runtime.error_or

val aft_loss_param_to_yojson : aft_loss_param -> Yojson.Safe.t

val aft_loss_param_of_yojson : 
  Yojson.Safe.t ->
  aft_loss_param Ppx_deriving_yojson_runtime.error_or

val softmax_multiclass_param_to_yojson : 
  softmax_multiclass_param ->
  Yojson.Safe.t

val softmax_multiclass_param_of_yojson : 
  Yojson.Safe.t ->
  softmax_multiclass_param Ppx_deriving_yojson_runtime.error_or

val lambda_rank_param_to_yojson : lambda_rank_param -> Yojson.Safe.t

val lambda_rank_param_of_yojson : 
  Yojson.Safe.t ->
  lambda_rank_param Ppx_deriving_yojson_runtime.error_or

val tree_to_yojson : tree -> Yojson.Safe.t

val tree_of_yojson : Yojson.Safe.t -> tree Ppx_deriving_yojson_runtime.error_or

val gbtree_to_yojson : gbtree -> Yojson.Safe.t

val gbtree_of_yojson : 
  Yojson.Safe.t ->
  gbtree Ppx_deriving_yojson_runtime.error_or

val gblinear_to_yojson : gblinear -> Yojson.Safe.t

val gblinear_of_yojson : 
  Yojson.Safe.t ->
  gblinear Ppx_deriving_yojson_runtime.error_or

val dart_to_yojson : dart -> Yojson.Safe.t

val dart_of_yojson : Yojson.Safe.t -> dart Ppx_deriving_yojson_runtime.error_or

val learner_model_param_to_yojson : learner_model_param -> Yojson.Safe.t

val learner_model_param_of_yojson : 
  Yojson.Safe.t ->
  learner_model_param Ppx_deriving_yojson_runtime.error_or

type gradient_booster =

| Gbtree of gbtree
| Gblinear of gblinear
| Dart of dart

val pp_gradient_booster : 
  Ppx_deriving_runtime.Format.formatter ->
  gradient_booster ->
  Ppx_deriving_runtime.unit

val show_gradient_booster : gradient_booster -> Ppx_deriving_runtime.string

val gradient_booster_of_yojson : 
  Yojson.Safe.t ->
  gradient_booster Ppx_deriving_yojson_runtime.error_or

val gradient_booster_to_yojson : 
  gradient_booster ->
  [> `Assoc of (string * Yojson.Safe.t) list ]

type objective =

| Reg_squarederror of reg_loss_param
| Reg_pseudohubererror of reg_loss_param
| Reg_squaredlogerror of reg_loss_param
| Reg_linear of reg_loss_param
| Binary_logistic of reg_loss_param

val pp_objective : 
  Ppx_deriving_runtime.Format.formatter ->
  objective ->
  Ppx_deriving_runtime.unit

val show_objective : objective -> Ppx_deriving_runtime.string

val objective_of_yojson : 
  Yojson.Safe.t ->
  objective Ppx_deriving_yojson_runtime.error_or

val objective_to_yojson : 
  objective ->
  [> `Assoc of (string * Yojson.Safe.t) list ]

type learner = {

feature_names : string array;
feature_types : string array;
gradient_booster : gradient_booster;
objective : objective;
learner_model_param : learner_model_param;

}

and t = {

version : int * int * int;
learner : learner;

}

val pp_learner : 
  Ppx_deriving_runtime.Format.formatter ->
  learner ->
  Ppx_deriving_runtime.unit

val show_learner : learner -> Ppx_deriving_runtime.string

val pp : 
  Ppx_deriving_runtime.Format.formatter ->
  t ->
  Ppx_deriving_runtime.unit

val show : t -> Ppx_deriving_runtime.string

val learner_to_yojson : learner -> Yojson.Safe.t

val learner_of_yojson : 
  Yojson.Safe.t ->
  learner Ppx_deriving_yojson_runtime.error_or

val to_yojson : t -> Yojson.Safe.t

val of_yojson : Yojson.Safe.t -> t Ppx_deriving_yojson_runtime.error_or

val _ : Yojson.Safe.t -> t Ppx_deriving_yojson_runtime.error_or

package caisar-xgboost