Added classify method for perceptron

This commit is contained in:
Niklas Birk
2019-06-27 00:11:59 +02:00
parent 05dba5bae5
commit 757eb4dd2b
2 changed files with 32 additions and 2 deletions

View File

@ -1,6 +1,7 @@
package machine_learning.perceptron;
import machine_learning.Vector;
import machine_learning.nearest_neighbour.DataClass;
import java.util.List;
@ -42,6 +43,11 @@ public class Perceptron
System.out.println("-----------------------------------------------------------------");
}
public DataClass classify(Vector vector)
{
return this.weight.scalar(vector) > 0 ? DataClass.POSITIVE : DataClass.NEGATIVE;
}
private Vector getInitializationVector(List<Vector> positives, List<Vector> negatives)
{
var a = new Vector(positives.get(0).dimension());