0

R h2oディープラーニングパッケージでパラメータの貢献の各パーセンテージを取得するにはどうすればよいですか?

library(h2o)
localH2O = h2o.init(ip = "localhost", port = 54321, startH 2O = TRUE)
irisPath = system.file("extdata", "iris.csv", package = "h2o")
iris.hex = h2o.importFile(localH2O, path = irisPath)
h2o.deeplearning(x = 1:4, y = 5, data = iris.hex, activation = "Tanh")
h2o.shutdown(localH2O)
4

1 に答える 1

1

モデルを構築するときに、次の条件を追加します。variable_importance = T

これにより、モデルが構築されたときに、変数の重要度が返されます。

R のディープ ラーニング デモでは、モデル構築プロセスを変更する必要があります。まず、次のコードを実行してデモを起動します。

library(h2o)
conn <- h2o.init(nthreads = -1)
demo(h2o.deeplearning)

次に、前述の条件を追加して、モデルのビルドを開始するコードを調整します。

model = h2o.deeplearning(x = setdiff(colnames(prostate.hex), c("ID","CAPSULE")), y = "CAPSULE", training_frame = prostate.hex, activation = "Tanh", hidden = c(10, 10, 10), epochs = 10000, variable_importances = T)

最後に、次のようにして変数の重要度を取得できます。

> h2o.varimp(model)
Variable Importances:
  variable relative_importance scaled_importance percentage
1      PSA            1.000000          1.000000   0.175660
2      VOL            0.937293          0.937293   0.164645
3  GLEASON            0.930565          0.930565   0.163463
4      AGE            0.799607          0.799607   0.140459
5    DCAPS            0.793741          0.793741   0.139429
6    DPROS            0.703781          0.703781   0.123626
7     RACE            0.527824          0.527824   0.092718

お役に立てれば！

于 2015-08-14T01:05:53.897 に答える