Inheritance: implements Preprocessor
Example #1
0
 public function testImputerWorksOnFitSamples()
 {
     $trainData = [[1, 3, 4], [6, 7, 8], [8, 7, 5]];
     $data = [[1, 3, null], [6, null, 8], [null, 7, 5]];
     $imputeData = [[1, 3, 5.66], [6, 5.66, 8], [5, 7, 5]];
     $imputer = new Imputer(null, new MeanStrategy(), Imputer::AXIS_COLUMN, $trainData);
     $imputer->transform($data);
     $this->assertEquals($imputeData, $data, '', $delta = 0.01);
 }