91 lines
2.1 KiB
PHP
91 lines
2.1 KiB
PHP
<?php
|
|
|
|
namespace Rubix\ML\Benchmarks\Datasets;
|
|
|
|
use Tensor\Vector;
|
|
use Rubix\ML\Datasets\Generators\Blob;
|
|
use Rubix\ML\Datasets\Generators\Agglomerate;
|
|
|
|
/**
|
|
* @Groups({"Datasets"})
|
|
* @BeforeMethods({"setUp"})
|
|
*/
|
|
class LabeledBench
|
|
{
|
|
protected const DATASET_SIZE = 25000;
|
|
|
|
protected const SUBSAMPLE_SIZE = 10000;
|
|
|
|
/**
|
|
* @var \Rubix\ML\Datasets\Labeled;
|
|
*/
|
|
protected $dataset;
|
|
|
|
/**
|
|
* @var float[]
|
|
*/
|
|
protected $weights;
|
|
|
|
public function setUp() : void
|
|
{
|
|
$generator = new Agglomerate([
|
|
'Iris-setosa' => new Blob([5.0, 3.42, 1.46, 0.24], [0.35, 0.38, 0.17, 0.1]),
|
|
'Iris-versicolor' => new Blob([5.94, 2.77, 4.26, 1.33], [0.51, 0.31, 0.47, 0.2]),
|
|
'Iris-virginica' => new Blob([6.59, 2.97, 5.55, 2.03], [0.63, 0.32, 0.55, 0.27]),
|
|
]);
|
|
|
|
$this->dataset = $generator->generate(self::DATASET_SIZE);
|
|
|
|
$this->weights = Vector::rand($this->dataset->numRows())->asArray();
|
|
}
|
|
|
|
/**
|
|
* @Subject
|
|
* @Iterations(5)
|
|
* @OutputTimeUnit("milliseconds", precision=3)
|
|
*/
|
|
public function randomSubset() : void
|
|
{
|
|
$this->dataset->randomSubset(self::SUBSAMPLE_SIZE);
|
|
}
|
|
|
|
/**
|
|
* @Subject
|
|
* @Iterations(5)
|
|
* @OutputTimeUnit("milliseconds", precision=3)
|
|
*/
|
|
public function randomSubsetWithReplacement() : void
|
|
{
|
|
$this->dataset->randomSubsetWithReplacement(self::SUBSAMPLE_SIZE);
|
|
}
|
|
|
|
/**
|
|
* @Subject
|
|
* @Iterations(5)
|
|
* @OutputTimeUnit("milliseconds", precision=3)
|
|
*/
|
|
public function randomWeightedSubsetWithReplacement() : void
|
|
{
|
|
$this->dataset->randomWeightedSubsetWithReplacement(self::SUBSAMPLE_SIZE, $this->weights);
|
|
}
|
|
|
|
/**
|
|
* @Subject
|
|
* @Iterations(5)
|
|
* @OutputTimeUnit("milliseconds", precision=3)
|
|
*/
|
|
public function splitByColumn() : void
|
|
{
|
|
$this->dataset->splitByColumn(2, 3.0);
|
|
}
|
|
|
|
/**
|
|
* @Subject
|
|
* @Iterations(5)
|
|
* @OutputTimeUnit("milliseconds", precision=3)
|
|
*/
|
|
public function dropColumns() : void
|
|
{
|
|
$this->dataset->dropColumns([1, 2]);
|
|
}
|
|
}
|