Skip to content

Instantly share code, notes, and snippets.

@YohanObadia

YohanObadia/knn_impute_example.py

Created February 4, 2017 13:00
Show Gist options
  • Save YohanObadia/a3e568cd7e3c57228e474b5f1608c27e to your computer and use it in GitHub Desktop.
Save YohanObadia/a3e568cd7e3c57228e474b5f1608c27e to your computer and use it in GitHub Desktop.
Download ZIP
Example of use for the knn_impute function
Raw
knn_impute_example.py
knn_impute(target=df['Age'], attributes=df.drop(['Age', 'PassengerId'], 1),
aggregation_method="median", k_neighbors=10, numeric_distance='euclidean',
categorical_distance='hamming', missing_neighbors_threshold=0.8)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment