我正在尝试使用Plumber和Swagger将一个R机器学习分类模型暴露为Web服务。我已经训练了一个模型并将其保存为“j48.model.rda”。现在我在一个名为“myFile.R”的文件中加载这个模型。该文件包含以下R代码:
library(rJava)jsDirData <- "C:/AA Research/Playpen/Data"setwd(jsDirData)#加载保存的模型load(file="j48.model.rda", envir = parent.frame(), verbose = FALSE) #' @param naasra90th:numeric 段的90th百分位Naasra值#' @param rut90th:numeric 段的90th百分位Rut深度#' @param surfAge:numeric 段的表面年龄,单位为年(可以是小数)#' @param rutRate90th:numeric 90th百分位Rut深度的Rut率(毫米/年)#' @param maintCount:int 维护操作的数量#' @get /getTreatment#' @html#' @response 200 返回j48模型的治疗类别预测(ThinAC或none)#' @default Bonk!getTreatment <- function(naasra90th, rut90th, surfAge, rutRate90th, maintCount) { xVals <- list(naasra90th = naasra90th, rut90th = rut90th, surfAge = surfAge, rutRate90th = rutRate90th, maintCount = maintCount) nData <- as.data.frame(xVals) pred <- predict(j48.model,newdata = nData) res <- as.character(pred) return(res)}t <- getTreatment(50,8.8,5,0.3,0) #应该返回"none"t #"none" 正确!t <- getTreatment(888,888,888,888,888) #应该返回"ThinAC"t #"ThinAC" 正确!
如您所见,从最后几行可以看出,当我在R-Studio中直接调用该函数时,它给出了正确的分类。但是现在我尝试通过Plumber/Swagger Web服务调用这个方法,如下所示:
library(plumber)jsDirData <- "C:/AA Research/Playpen/Data"setwd(jsDirData)r <- plumb("myfile.R")r$run(port=8000)
当我运行这段代码时,Swagger会打开浏览器并正确显示API。然而,当我使用“Try It”按钮测试API时,无论我传入什么参数,它总是显示结果为“none”。例如,如果我输入与上述第二个方法调用相同的参数集(即所有参数均为888),那么它返回“none”,而应该返回“ThinAC”。
我哪里做错了?
回答:
我认为您从Swagger调用中接收到的值仍然是字符类,因为Plumber不会对查询字符串参数进行任何转换。
在执行as.data.frame
之前,尝试更改xVals中值的类别
xVals <- lapply(xVals, as.numeric)
为了验证这个假设,您可以在as.data.frame
之后插入browser()
,并使用lapply(nData, class)
检查nData
中值的类别。
祝您好运