From 520633ae8c0ab6d02266fbcbc3b1e6004f63f68d Mon Sep 17 00:00:00 2001 From: srzeszut Date: Sun, 20 Oct 2024 15:58:05 +0200 Subject: [PATCH] fix doctest --- lib/scholar/impute/knn_imputer.ex | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/scholar/impute/knn_imputer.ex b/lib/scholar/impute/knn_imputer.ex index dcc716bf..0d602dc5 100644 --- a/lib/scholar/impute/knn_imputer.ex +++ b/lib/scholar/impute/knn_imputer.ex @@ -121,7 +121,7 @@ defmodule Scholar.Impute.KNNImputer do ## Examples iex> x = Nx.tensor([[40.0, 2.0],[4.0, 5.0],[7.0, :nan],[:nan, 8.0],[11.0, 11.0]]) - iex> imputer = Scholar.Impute.KNNImputer.fit(x, strategy: :mean) + iex> imputer = Scholar.Impute.KNNImputer.fit(x, number_of_neighbors: 2) iex> Scholar.Impute.KNNImputer.transform(imputer, x) Nx.tensor( [