search

ml-lib / CodeLib

Code library for common machine learning algorithms
BSD 3-Clause "New" or "Revised" License
2 stars 1 forks source link

[Feature]: kNN: Optimal k #3

Closed bdiptesh closed 2 years ago

bdiptesh commented 2 years ago

Is your feature request related to a problem? Please describe.

kNN module.

Describe the solution you'd like

kNN module for both classification and regression. Determine optimal k.

Methods:

  1. Classification
  2. Regression

Expected input(s)

df: pandas.DataFrame
y_var: List[str]
x_var: List[str]
method: Union[str]
param: Dict

Expected output(s)

fit
predict

Additional context

No response

Acceptance criteria

Version

v0.4.0