From 04244d3f791711b614d962e64d00263a094f88ba Mon Sep 17 00:00:00 2001
From: Rafael Escaleira <rafaelescaleira@icloud.com>
Date: Tue, 11 Jun 2024 18:56:32 -0400
Subject: [PATCH] Update LinearRegression.swift

---
 Sources/RefdsAlgorithm/MachineLearning/LinearRegression.swift | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Sources/RefdsAlgorithm/MachineLearning/LinearRegression.swift b/Sources/RefdsAlgorithm/MachineLearning/LinearRegression.swift
index 1252662..f5a128d 100644
--- a/Sources/RefdsAlgorithm/MachineLearning/LinearRegression.swift
+++ b/Sources/RefdsAlgorithm/MachineLearning/LinearRegression.swift
@@ -78,7 +78,7 @@ public class LinearRegression {
         semaphore.signal()
     }
 
-    public func predict(for features: [Double]) -> Double? {
+    public func predict(for features: [NSNumber]) -> Double? {
         semaphore.wait()
         guard !coefficients.isEmpty,
               var prediction = coefficients[safe: 0]
@@ -86,7 +86,7 @@ public class LinearRegression {
         
         for i in features.indices {
             if let coefficient = coefficients[safe: i + 1] {
-                prediction += coefficient * features[i]
+                prediction += coefficient * features[i].doubleValue
             }
         }
         semaphore.signal()