Commit b00c4111 authored by leswing's avatar leswing
Browse files

Apparently ordering matters

parent 2bb0125e
Loading
Loading
Loading
Loading
+8 −5
Original line number Diff line number Diff line
@@ -27,9 +27,12 @@ class SklearnModel(Model):
  Abstract base class for different ML models.
  """

  def __init__(self, use_weights=True, **kwargs):
  def __init__(self, **kwargs):
    super(SklearnModel, self).__init__(**kwargs)
    self.use_weights = use_weights
    if 'use_weights' in kwargs:
      self.use_weights = kwargs['use_weights']
    else:
      self.use_weights = True
    for model_instance in NON_WEIGHTED_MODELS:
      if isinstance(self.model_instance, model_instance):
        self.use_weights = False