///|
fn main {
  let dataset = @tree.classification_dataset(
    [[0.1, 1.0], [0.3, 0.9], [1.8, 0.2], [2.0, 0.1]],
    [0, 0, 1, 1],
    2,
  ).unwrap()
  let model = @tree.train_classifier(
    dataset,
    @tree.ClassificationConfig::default(),
  ).unwrap()
  let predictions = model.predict_batch([[0.2, 0.8], [1.9, 0.2]]).unwrap()
  println("first=\{predictions[0]}, second=\{predictions[1]}")
  let probabilities = model.predict_proba([0.2, 0.8]).unwrap()
  println("class0=\{probabilities[0]}, class1=\{probabilities[1]}")
}