56 lines
1.6 KiB
PHP
56 lines
1.6 KiB
PHP
<?php
|
|
|
|
namespace App\Services\IterationEventBuffer;
|
|
|
|
class PerceptronLimitedEpochEventBuffer implements IPerceptronIterationEventBuffer
|
|
{
|
|
private array $data;
|
|
|
|
private int $underSizeIncreaseCount = 0;
|
|
|
|
public function __construct(
|
|
private string $sessionId,
|
|
private string $trainingId,
|
|
private int $epochInterval,
|
|
private int $sizeIncreaseStart = 10,
|
|
) {
|
|
$this->data = [];
|
|
}
|
|
|
|
public function flush(): void
|
|
{
|
|
event(new \App\Events\PerceptronTrainingIteration($this->data, $this->sessionId, $this->trainingId));
|
|
$this->data = [];
|
|
}
|
|
|
|
public function addIteration(int $epoch, int $exampleIndex, float $error, array $synaptic_weights): void
|
|
{
|
|
$newData = [
|
|
'epoch' => $epoch,
|
|
'exampleIndex' => $exampleIndex,
|
|
'error' => $error,
|
|
'weights' => $synaptic_weights,
|
|
];
|
|
|
|
if ($this->underSizeIncreaseCount <= $this->sizeIncreaseStart) { // Special case where we need to send each iteration separately
|
|
$this->underSizeIncreaseCount++;
|
|
$this->data[] = $newData;
|
|
$this->flush();
|
|
|
|
return;
|
|
}
|
|
|
|
$lastEpoch = $this->data[0]['epoch'] ?? null;
|
|
if ($this->data && $lastEpoch !== $epoch) { // Current Epoch has changed from the last one
|
|
if ($lastEpoch % $this->epochInterval === 0) { // The last epoch need to be sent
|
|
$this->flush(); // Flush all data from the previous epoch
|
|
} else {
|
|
$this->data = [];
|
|
}
|
|
|
|
$lastEpoch = $epoch;
|
|
}
|
|
$this->data[] = $newData;
|
|
}
|
|
}
|