| 135 | |
| 136 | |
| 137 | class parameter(Structure): |
| 138 | _names = ["solver_type", "eps", "C", "nr_weight", "weight_label", "weight", "p"] |
| 139 | _types = [c_int, c_double, c_double, c_int, POINTER(c_int), POINTER(c_double), c_double] |
| 140 | _fields_ = genFields(_names, _types) |
| 141 | |
| 142 | def __init__(self, options = None): |
| 143 | if options == None: |
| 144 | options = '' |
| 145 | self.parse_options(options) |
| 146 | |
| 147 | def __str__(self): |
| 148 | s = '' |
| 149 | attrs = parameter._names + list(self.__dict__.keys()) |
| 150 | values = map(lambda attr: getattr(self, attr), attrs) |
| 151 | for attr, val in zip(attrs, values): |
| 152 | s += (' %s: %s\n' % (attr, val)) |
| 153 | s = s.strip() |
| 154 | |
| 155 | return s |
| 156 | |
| 157 | def set_to_default_values(self): |
| 158 | self.solver_type = L2R_L2LOSS_SVC_DUAL |
| 159 | self.eps = float('inf') |
| 160 | self.C = 1 |
| 161 | self.p = 0.1 |
| 162 | self.nr_weight = 0 |
| 163 | self.weight_label = (c_int * 0)() |
| 164 | self.weight = (c_double * 0)() |
| 165 | self.bias = -1 |
| 166 | self.cross_validation = False |
| 167 | self.nr_fold = 0 |
| 168 | self.print_func = None |
| 169 | |
| 170 | def parse_options(self, options): |
| 171 | if isinstance(options, list): |
| 172 | argv = options |
| 173 | elif isinstance(options, str): |
| 174 | argv = options.split() |
| 175 | else: |
| 176 | raise TypeError("arg 1 should be a list or a str.") |
| 177 | self.set_to_default_values() |
| 178 | self.print_func = cast(None, PRINT_STRING_FUN) |
| 179 | weight_label = [] |
| 180 | weight = [] |
| 181 | |
| 182 | i = 0 |
| 183 | while i < len(argv) : |
| 184 | if argv[i] == "-s": |
| 185 | i = i + 1 |
| 186 | self.solver_type = int(argv[i]) |
| 187 | elif argv[i] == "-c": |
| 188 | i = i + 1 |
| 189 | self.C = float(argv[i]) |
| 190 | elif argv[i] == "-p": |
| 191 | i = i + 1 |
| 192 | self.p = float(argv[i]) |
| 193 | elif argv[i] == "-e": |
| 194 | i = i + 1 |