From 03f4707ea0ce0beccdd403c33948f84ac0a4d463 Mon Sep 17 00:00:00 2001 From: Mateusz Date: Tue, 17 Oct 2023 23:51:30 +0200 Subject: [PATCH] Correct tests --- lib/scholar/linear/isotonic_regression.ex | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lib/scholar/linear/isotonic_regression.ex b/lib/scholar/linear/isotonic_regression.ex index 0360b995..46221c52 100644 --- a/lib/scholar/linear/isotonic_regression.ex +++ b/lib/scholar/linear/isotonic_regression.ex @@ -204,7 +204,7 @@ defmodule Scholar.Linear.IsotonicRegression do iex> Scholar.Linear.IsotonicRegression.predict(model, to_predict) #Nx.Tensor< f32[10] - [1.0, 1.6666667461395264, 2.3333334922790527, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0] + [1.0, 1.6666667461395264, 2.3333332538604736, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0] > """ defn predict(model, x) do @@ -259,7 +259,7 @@ defmodule Scholar.Linear.IsotonicRegression do ] ), x: Nx.tensor( - [1.0, 4.0, 7.0, 9.0, 10.0] + [1.0, 4.0, 7.0, 9.0, 10.0, 11.0] ) } }