R/nearestNeighborCleanup.R

In kaiyaprovost/subsppLabelR: Subspecies Labeled Occurences Generator

#' Clean points via nearest neighbors
#'
#' This function calculates unlabeled points based on their assigned
#' labels, assuming the assigned labels are correct.It calculates them
#' based on their location in latitude and longitude space.
#'
#' @param good_points XX
#' @param bad_points XX
#' @param k XX
#'
#' @export
#' @examples
#'
#' ## i am an example
#'
nearestNeighborCleanup = function(good_points,bad_points,k=21){
  pr <- class::knn(train=good_points[,c("longitude","latitude")],
                   test=bad_points[,c("longitude","latitude")],
                   cl=good_points[,c("assigned")],
                   k=k,
                   use.all=T,
                   prob=T)
  bad_points$KNN = pr
  bad_points$KNNprob = attributes(pr)$prob
  return(bad_points)
}