Работа с упорядоченными массивами в PHP-ML

Использую класс PHP-ML, есть массив данных, нужно по значению ключа sum, предугадать остальные значения ключей:

$data = array(
    array(
        "id" => 4,
        "number" => "H9142",
        "sum" => "2100000",
        "status" => "1",
        "appr" => "2",
        "payment_wod" => 53326,
        "max_sum_wod" => 1365000,
        "payment_wd" => 30129,
        "max_sum_wd" => 2100000,
        "total" => "12",
    ),
    array(
        "id" => 5,
        "number" => "55953",
        "sum" => "4000000",
        "status" => "1",
        "appr" => "2",
        "payment_wod" => 101574,
        "max_sum_wod" => 2200000,
        "payment_wd" => 57388,
        "max_sum_wd" => 6000000,
        "total" => "12",
    ),
);

Есть пример кода, где данные берутся из CSV, но для работы с массивом, судя по официальной документации PHP-ML - нужно использовать объект ArrayDataset, дальше понять не могу, как оформить ArrayDataset для текущей задачи.

<?php
require "vendor/autoload.php";

use Phpml\Dataset\CsvDataset;
use Phpml\Regression\LeastSquares;

$dataset = new CsvDataset('train-data.csv', 5, true);

$samples = $dataset->getSamples();
$targets = $dataset->getTargets();

$regression = new LeastSquares();
$regression->train($samples, $targets);
$predict = $regression->predict([
    [1940,3,3,2,1],
    [2420,4,3,4,0],
    [2150,3,3,5,1],
    [1980,2,2,2,0],
]);
print_r($predict);