Skip to content

Commit

Permalink
edit example documentation for KNN
Browse files Browse the repository at this point in the history
Signed-off-by: Minh Khue Tran <minh.khue.tran@fau.de>
  • Loading branch information
Minh Khue Tran committed Feb 4, 2025
1 parent f769af9 commit dbe1f48
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ class KNearestNeighbors(MachineLearningInterface):
knn.train(train_df)
predictions = knn.predict(test_df)
```
Parameters:
df (pyspark.sql.Dataframe): DataFrame containing the features and labels
features_col (str): Name of the column containing the features (the input). Default is 'features'
Expand Down

0 comments on commit dbe1f48

Please # to comment.