use of org.bytedeco.opencv.opencv_ml.SVM in project qupath by qupath.
the class SVMClassifier method createClassifier.
@Override
protected SVM createClassifier() {
SVM svm = SVM.create();
ParameterList params = getParameterList();
if (params != null) {
String kernel = (String) params.getChoiceParameterValue("kernel");
kernel = kernel.toLowerCase();
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;
}
Aggregations