131 lines
3.1 KiB
PHP
131 lines
3.1 KiB
PHP
<?php
|
|
|
|
namespace Rubix\ML\Tests\NeuralNet\ActivationFunctions;
|
|
|
|
use Tensor\Matrix;
|
|
use Rubix\ML\NeuralNet\ActivationFunctions\ThresholdedReLU;
|
|
use Rubix\ML\NeuralNet\ActivationFunctions\ActivationFunction;
|
|
use PHPUnit\Framework\TestCase;
|
|
use Generator;
|
|
|
|
/**
|
|
* @group ActivationFunctions
|
|
* @covers \Rubix\ML\NeuralNet\ActivationFunctions\ThresholdedReLU
|
|
*/
|
|
class ThresholdedReLUTest extends TestCase
|
|
{
|
|
/**
|
|
* @var \Rubix\ML\NeuralNet\ActivationFunctions\ThresholdedReLU
|
|
*/
|
|
protected $activationFn;
|
|
|
|
/**
|
|
* @before
|
|
*/
|
|
protected function setUp() : void
|
|
{
|
|
$this->activationFn = new ThresholdedReLU(0.1);
|
|
}
|
|
|
|
/**
|
|
* @test
|
|
*/
|
|
public function build() : void
|
|
{
|
|
$this->assertInstanceOf(ThresholdedReLU::class, $this->activationFn);
|
|
$this->assertInstanceOf(ActivationFunction::class, $this->activationFn);
|
|
}
|
|
|
|
/**
|
|
* @test
|
|
* @dataProvider computeProvider
|
|
*
|
|
* @param \Tensor\Matrix $input
|
|
* @param array[] $expected
|
|
*/
|
|
public function compute(Matrix $input, array $expected) : void
|
|
{
|
|
$activations = $this->activationFn->compute($input)->asArray();
|
|
|
|
$this->assertEquals($expected, $activations);
|
|
}
|
|
|
|
/**
|
|
* @return \Generator<array>
|
|
*/
|
|
public function computeProvider() : Generator
|
|
{
|
|
yield [
|
|
Matrix::quick([
|
|
[1.0, -0.5, 0.0, 20.0, -10.0],
|
|
]),
|
|
[
|
|
[1.0, 0.0, 0.0, 20.0, 0.0],
|
|
],
|
|
];
|
|
|
|
yield [
|
|
Matrix::quick([
|
|
[-0.12, 0.31, -0.49],
|
|
[0.99, 0.08, -0.03],
|
|
[0.05, -0.52, 0.54],
|
|
]),
|
|
[
|
|
[0.0, 0.31, 0.0],
|
|
[0.99, 0.0, 0.0],
|
|
[0.0, 0.0, 0.54],
|
|
],
|
|
];
|
|
}
|
|
|
|
/**
|
|
* @test
|
|
* @dataProvider differentiateProvider
|
|
*
|
|
* @param \Tensor\Matrix $input
|
|
* @param \Tensor\Matrix $activations
|
|
* @param array[] $expected
|
|
*/
|
|
public function differentiate(Matrix $input, Matrix $activations, array $expected) : void
|
|
{
|
|
$derivatives = $this->activationFn->differentiate($input, $activations)->asArray();
|
|
|
|
$this->assertEquals($expected, $derivatives);
|
|
}
|
|
|
|
/**
|
|
* @return \Generator<array>
|
|
*/
|
|
public function differentiateProvider() : Generator
|
|
{
|
|
yield [
|
|
Matrix::quick([
|
|
[1.0, -0.5, 0.0, 20.0, -10.0],
|
|
]),
|
|
Matrix::quick([
|
|
[1.0, 0.0, 0.0, 20.0, 0.0],
|
|
]),
|
|
[
|
|
[1, 0, 0, 1, 0],
|
|
],
|
|
];
|
|
|
|
yield [
|
|
Matrix::quick([
|
|
[-0.12, 0.31, -0.49],
|
|
[0.99, 0.08, -0.03],
|
|
[0.05, -0.52, 0.54],
|
|
]),
|
|
Matrix::quick([
|
|
[0.0, 0.31, 0.0],
|
|
[0.99, 0.0, 0.0],
|
|
[0.0, 0.0, 0.54],
|
|
]),
|
|
[
|
|
[0, 1, 0],
|
|
[1, 0, 0],
|
|
[0, 0, 1],
|
|
],
|
|
];
|
|
}
|
|
}
|