Fix backpropagation random error (#157)

This commit is contained in:
David Monllaó 2017-11-20 23:11:21 +01:00 committed by Arkadiusz Kondas
parent ff80af2044
commit 333598b472

View File

@ -112,16 +112,16 @@ class MLPClassifierTest extends TestCase
public function testBackpropagationLearningMultilayer(): void public function testBackpropagationLearningMultilayer(): void
{ {
// Multi-layer 2 classes. // Multi-layer 2 classes.
$network = new MLPClassifier(5, [3, 2], ['a', 'b']); $network = new MLPClassifier(5, [3, 2], ['a', 'b', 'c']);
$network->train( $network->train(
[[1, 0, 0, 0, 0], [0, 1, 1, 0, 0], [1, 1, 1, 1, 1], [0, 0, 0, 0, 0]], [[1, 0, 0, 0, 0], [0, 1, 1, 0, 0], [1, 1, 1, 1, 1], [0, 0, 0, 0, 0]],
['a', 'b', 'a', 'b'] ['a', 'b', 'a', 'c']
); );
$this->assertEquals('a', $network->predict([1, 0, 0, 0, 0])); $this->assertEquals('a', $network->predict([1, 0, 0, 0, 0]));
$this->assertEquals('b', $network->predict([0, 1, 1, 0, 0])); $this->assertEquals('b', $network->predict([0, 1, 1, 0, 0]));
$this->assertEquals('a', $network->predict([1, 1, 1, 1, 1])); $this->assertEquals('a', $network->predict([1, 1, 1, 1, 1]));
$this->assertEquals('b', $network->predict([0, 0, 0, 0, 0])); $this->assertEquals('c', $network->predict([0, 0, 0, 0, 0]));
} }
public function testBackpropagationLearningMulticlass(): void public function testBackpropagationLearningMulticlass(): void