R/aif_utils.R

In SSaishruthi/raif360-install:

#' Function to create TensorFlow session
#' @export
#'
tf_session <-function(){
  tf$Session()
}

#' Function to close TensorFlow session
#' @param sess tensorflow session
#' @export
#'
tf_close <-function(sess){
  (sess$close())
}
#' Function to reset TensorFlow default graph
#' @export
#'
tf_reset_default_graph <-function(){
  tf$reset_default_graph()
}

setGeneric("fit",function(object, dataset, ...){
  fit(object, dataset, ...)}
)

setGeneric("transform",function(object,dataset, ...){
  transform(object,dataset, ...)}
)

setGeneric("fit_predict",function(object,dataset,...){
  fit_obj <- fit(object, dataset, ...)
  predict(fit_obj,dataset, ...)}
)

setGeneric("fit_transform",function(object,dataset, ...){
  fit_obj <- fit(object, dataset, ...)
  transform(fit_obj,dataset)}
)

#' Function to convert lists of lists to python dictionarys
#' @param i The list to be converted
#' @export
#' @importFrom reticulate py_dict
#'
py_dict_conv <- function(i){
  py_dict(i[[1]],i[[2]], convert = FALSE)
}