lib/eps/evaluators/linear_regression.rb in eps-0.3.0 vs lib/eps/evaluators/linear_regression.rb in eps-0.3.1
- old
+ new
@@ -8,10 +8,10 @@
@features = features
@text_features = text_features || {}
end
def predict(x)
- intercept = @coefficients["_intercept"]
+ intercept = @coefficients["_intercept"] || 0.0
scores = [intercept] * x.size
@features.each do |k, type|
raise "Missing data in #{k}" if !x.columns[k] || x.columns[k].any?(&:nil?)