R/splitters.R

Defines functions parent_label GrandparentSplitter

Documented in GrandparentSplitter parent_label

#' @title GrandparentSplitter
#'
#' @description Split `items` from the grand parent folder names (`train_name` and `valid_name`).
#'
#'
#' @param train_name train folder name
#' @param valid_name validation folder name
#' @return None
#' @export
GrandparentSplitter <- function(train_name = "train", valid_name = "valid") {

  fastai2$vision$all$GrandparentSplitter(
    train_name = train_name,
    valid_name = valid_name
  )

}


#' @title Parent_label
#'
#' @description Label `item` with the parent folder name.
#'
#'
#' @param o string, dir path
#' @return vector
#' @export
parent_label <- function(o) {

  if(missing(o)) {
    fastai2$vision$all$parent_label
  } else {
    fastai2$vision$all$parent_label(
      o = o
    )
  }


}

Try the fastai package in your browser

Any scripts or data that you put into this service are public.

fastai documentation built on March 31, 2023, 11:41 p.m.