2

にトレーニング済みのxgboostモデルがRあり、アプリケーション サーバーで R を実行できますが、インストールできませんpackage:xgboost(R 用の JVM ベースのインタープリターである renjin ではサポートされていません)。そのため、モデルをデプロイするには、新しいデータをスコアリングせずにスコアリングする方法が必要でしたxgb.predict。で作成されたデータ フレームを使用して、R でスコアリング関数を作成しましたxgb.model.dt.tree。これは正常に動作しますが、非常に低速です。1 つのレコードを獲得するのに 7 秒かかります。どうすればコードをより速く実行できるか疑問に思っています。助けていただければ幸いです。

私は 2 つの for ループと 1 つの while ループを使用していますが、これは R では明らかに非常に遅いです。

Rcppオプションのようですが、C++を知りません。

これが私のコードです。18クラス（0、500、1000など）のマルチクラス分類。モデルnrounds=50はトータル18*50=900ブースターで構築されています。ツリーの深さは 8 に設定され、ダンプ データ フレームaは size238252 X 10です。

fun_score <- function (testsample) {
  df <- data.frame('0'=NA,'500'=NA,'1000'=NA,'1500'=NA,'2000'=NA,'3000'=NA,'5000'=NA,'6000'=NA,'8000'=NA,
                   '10000'=NA,'15000'=NA,'20000'=NA,'25000'=NA,'40000'=NA,'50000'=NA,'70000'=NA,'77000'=NA,'120000'=NA)
for (i in 0:17) { 
    twght <- 0
    for(tree_num in seq(i,899,18)){
      tr=a[Tree==tree_num]
      rid <- which(tr[,2]==0)
      splitvar <- as.character(tr[rid,4])
      while (grepl("Leaf",tr[rid,4])==F) {
        next_split <- as.character(ifelse(is.null(testsample[splitvar]), tr[rid,7],
                                          ifelse( as.numeric(testsample[splitvar]) < tr[rid,5], tr[rid,6],tr[rid,7])))
        rid <- which(tr[,3] == next_split)
        splitvar <- as.character(tr[rid,4])
      }
      w=(tr[rid,9])
      twght=twght+w
    }
    twght=twght+0.5 # bias 0.5
    df[,i+1]=twght
}
  df = as.data.frame(t(apply(df, 1, function(x)(exp(x))/(sum(exp(x))))))
  df$class=substr(colnames(df)[apply(df,1,which.max)],2, nchar(colnames(df)[apply(df,1,which.max)]))
  return(df)
}

そして、ツリーdata.frameは次のようになります

> head(a)
   Tree Node  ID Feature     Split  Yes   No Missing    Quality     Cover
1:    0    0 0-0     DBR 14.900000  0-1  0-2     0-1 31579.3008 28727.481
2:    0    1 0-1     DBR  8.815001  0-3  0-4     0-3  4707.3477 15235.673
3:    0    2 0-2     DBR 20.584999  0-5  0-6     0-5  3631.3484 13491.809
4:    0    3 0-3    DSLR 18.500000  0-7  0-8     0-7  1703.1209  6624.963
5:    0    4 0-4     RAD -0.500000  0-9 0-10     0-9  1390.2230  8610.710
6:    0    5 0-5    DSLR 27.500000 0-11 0-12    0-11   861.4305  4966.624
4

1 に答える 1