| 3 | |
| 4 | |
| 5 | class SVMEvaluator(BaseSKLearnEvaluator): |
| 6 | def __init__(self, linear=True, params=None): |
| 7 | if linear: |
| 8 | self.evaluator = LinearSVC() |
| 9 | else: |
| 10 | self.evaluator = SVC() |
| 11 | if params is None: |
| 12 | params = {'C': [0.001, 0.01, 0.1, 1, 10, 100, 1000]} |
| 13 | super(SVMEvaluator, self).__init__(self.evaluator, params) |