Example #1
0
 public function testGuess()
 {
     $schema = new Schema();
     $schema->addField('rooms')->addField('area');
     $dataset = new KNN(3, $schema);
     $dataset->add(new DistanceNode(array('rooms' => 1, 'area' => 350, 'type' => 'apartment')));
     $dataset->add(new DistanceNode(array('rooms' => 2, 'area' => 300, 'type' => 'apartment')));
     $dataset->add(new DistanceNode(array('rooms' => 3, 'area' => 300, 'type' => 'apartment')));
     $dataset->add(new DistanceNode(array('rooms' => 4, 'area' => 250, 'type' => 'apartment')));
     $dataset->add(new DistanceNode(array('rooms' => 7, 'area' => 850, 'type' => 'house')));
     $dataset->add(new DistanceNode(array('rooms' => 7, 'area' => 900, 'type' => 'house')));
     $dataset->add(new DistanceNode(array('rooms' => 7, 'area' => 1200, 'type' => 'house')));
     $dataset->add(new DistanceNode(array('rooms' => 8, 'area' => 1500, 'type' => 'house')));
     $dataset->add(new DistanceNode(array('rooms' => 1, 'area' => 800, 'type' => 'flat')));
     $dataset->add(new DistanceNode(array('rooms' => 3, 'area' => 900, 'type' => 'flat')));
     $dataset->add(new DistanceNode(array('rooms' => 2, 'area' => 700, 'type' => 'flat')));
     $dataset->add(new DistanceNode(array('rooms' => 1, 'area' => 900, 'type' => 'flat')));
     $this->assertEquals('flat', $dataset->guess(new DistanceNode(array('rooms' => 4, 'area' => 900)), 'type'));
     $this->assertEquals('house', $dataset->guess(new DistanceNode(array('rooms' => 7, 'area' => 900)), 'type'));
     $this->assertEquals('apartment', $dataset->guess(new DistanceNode(array('rooms' => 1, 'area' => 200)), 'type'));
 }