Example usage for org.opencv.ml SVM setKernel

List of usage examples for org.opencv.ml SVM setKernel

Introduction

In this page you can find the example usage for org.opencv.ml SVM setKernel.

Prototype

public void setKernel(int kernelType) 

Source Link

Usage

From source file:qupath.opencv.classify.SVMClassifier.java

License:Open Source License

@Override
protected SVM createClassifier() {
    SVM svm = SVM.create();
    ParameterList params = getParameterList();
    //      System.out.println("Default C: " + svm.getC());

    if (params != null) {
        String kernel = (String) params.getChoiceParameterValue("kernel");
        kernel = kernel.toLowerCase();//w w  w .  jav  a  2s .  c om
        if (kernel.equals("linear"))
            svm.setKernel(SVM.LINEAR);
        else if (kernel.equals("polynomial"))
            svm.setKernel(SVM.POLY);
        else if (kernel.equals("rbf"))
            svm.setKernel(SVM.RBF);
        else if (kernel.equals("sigmoid"))
            svm.setKernel(SVM.SIGMOID);
        else if (kernel.equals("chi2"))
            svm.setKernel(SVM.CHI2);
        else if (kernel.equals("histogram intersection"))
            svm.setKernel(SVM.INTER);

        double c = params.getDoubleParameterValue("c");
        if (c > 0)
            svm.setC(c);
        svm.setGamma(params.getDoubleParameterValue("gamma"));
        svm.setDegree(params.getIntParameterValue("degree"));
    }
    return svm;
}