Added classify method for perceptron
This commit is contained in:
@ -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());
|
||||
|
Reference in New Issue
Block a user