mirror of
https://github.com/titanscouting/tra-analysis.git
synced 2024-11-13 22:56:18 +00:00
79 lines
2.4 KiB
Python
79 lines
2.4 KiB
Python
# Titan Robotics Team 2022: SVM submodule
|
|
# Written by Arthur Lu
|
|
# Notes:
|
|
# this should be imported as a python module using 'from tra_analysis import SVM'
|
|
# setup:
|
|
|
|
__version__ = "1.0.0"
|
|
|
|
__changelog__ = """changelog:
|
|
1.0.0:
|
|
- ported analysis.SVM() here
|
|
"""
|
|
|
|
__author__ = (
|
|
"Arthur Lu <learthurgo@gmail.com>",
|
|
)
|
|
|
|
__all__ = [
|
|
]
|
|
|
|
import sklearn
|
|
from sklearn import svm
|
|
from . import ClassificationMetric, RegressionMetric
|
|
|
|
class SVM:
|
|
|
|
class CustomKernel:
|
|
|
|
def __new__(cls, C, kernel, degre, gamma, coef0, shrinking, probability, tol, cache_size, class_weight, verbose, max_iter, decision_function_shape, random_state):
|
|
|
|
return sklearn.svm.SVC(C = C, kernel = kernel, gamma = gamma, coef0 = coef0, shrinking = shrinking, probability = probability, tol = tol, cache_size = cache_size, class_weight = class_weight, verbose = verbose, max_iter = max_iter, decision_function_shape = decision_function_shape, random_state = random_state)
|
|
|
|
class StandardKernel:
|
|
|
|
def __new__(cls, kernel, C=1.0, degree=3, gamma='auto_deprecated', coef0=0.0, shrinking=True, probability=False, tol=0.001, cache_size=200, class_weight=None, verbose=False, max_iter=-1, decision_function_shape='ovr', random_state=None):
|
|
|
|
return sklearn.svm.SVC(C = C, kernel = kernel, gamma = gamma, coef0 = coef0, shrinking = shrinking, probability = probability, tol = tol, cache_size = cache_size, class_weight = class_weight, verbose = verbose, max_iter = max_iter, decision_function_shape = decision_function_shape, random_state = random_state)
|
|
|
|
class PrebuiltKernel:
|
|
|
|
class Linear:
|
|
|
|
def __new__(cls):
|
|
|
|
return sklearn.svm.SVC(kernel = 'linear')
|
|
|
|
class Polynomial:
|
|
|
|
def __new__(cls, power, r_bias):
|
|
|
|
return sklearn.svm.SVC(kernel = 'polynomial', degree = power, coef0 = r_bias)
|
|
|
|
class RBF:
|
|
|
|
def __new__(cls, gamma):
|
|
|
|
return sklearn.svm.SVC(kernel = 'rbf', gamma = gamma)
|
|
|
|
class Sigmoid:
|
|
|
|
def __new__(cls, r_bias):
|
|
|
|
return sklearn.svm.SVC(kernel = 'sigmoid', coef0 = r_bias)
|
|
|
|
def fit(self, kernel, train_data, train_outputs): # expects *2d data, 1d labels or outputs
|
|
|
|
return kernel.fit(train_data, train_outputs)
|
|
|
|
def eval_classification(self, kernel, test_data, test_outputs):
|
|
|
|
predictions = kernel.predict(test_data)
|
|
|
|
return ClassificationMetric(predictions, test_outputs)
|
|
|
|
def eval_regression(self, kernel, test_data, test_outputs):
|
|
|
|
predictions = kernel.predict(test_data)
|
|
|
|
return RegressionMetric(predictions, test_outputs) |