Mojo struct

KNN

@memory_only
struct KNN

Classifier implementing the k-nearest neighbors vote.

Aliases

• MODEL_ID = 4
• metric_ids = List(String("euc"), String("man"), __list_literal__=NoneType(None))

Fields

• k (Int): Number of neighbors to use.
• metric (String): Metric to use for distance computation: Euclidean -> 'euc'; Manhattan -> 'man'.
• search_depth (Int): Current KDTree implementation applies some approximation to its search results. Increasing search_depth can lead to more accurate results at the cost of performance.
• kdtree (KDTree[True])
• y_train (Matrix)

Implemented traits

AnyType, CV, Copyable, Deinitable, Movable

Methods

__init__

fn def __init__(out self, k: Int = Int(3), metric: String = "euc", search_depth: Int = Int(1))

Args:

• k (Int)
• metric (String)
• search_depth (Int)
• self (Self)

Returns:

Self

Raises:

fn def __init__(out self, params: Dict[String, String])

Args:

• params (Dict[String, String])
• self (Self)

Returns:

Self

Raises:

fit

fn def fit(mut self, X: Matrix, y: Matrix)

Fit the k-nearest neighbors classifier from the training dataset.

Args:

• self (Self)
• X (Matrix)
• y (Matrix)

Raises:

predict

fn def predict(self, X: Matrix) -> Matrix

Predict the class indices for the provided data.

Args:

• self (Self)
• X (Matrix)

Returns:

Matrix: Class indices for each data sample.

Raises:

save

fn def save(self, path: String)

Save model data necessary for prediction to the specified path.

Args:

• self (Self)
• path (String)

Raises:

load

@staticmethod
fn def load(path: String) -> Self

Load a saved model from the specified path for prediction.

Args:

• path (String)

Returns:

Self

Raises: