GBM 모델의 각 트리에서 사용 된 분할을 보려면 어떻게해야합니까?
감사
GBM 모델의 각 트리에서 사용 된 분할을 보려면 어떻게해야합니까?
감사
당신은 h2o.download_pojo()
와 POJO로 모델을 내 보낸 다음 파일의 각 트리의 자세한 내용을 볼 수 있습니다.
H2O-3 MOJO 모델에 대한 시각화를 만드는 도구가 있습니다.
사용 R MOJO 만들고 다운로드 :
library(h2o)
h2o.init()
df <- h2o.importFile("http://s3.amazonaws.com/h2o-public-test-data/smalldata/airlines/allyears2k_headers.zip")
model <- h2o.gbm(model_id = "model",
training_frame = df,
x = c("Year", "Month", "DayofMonth", "DayOfWeek", "UniqueCarrier"),
y = "IsDepDelayed",
max_depth = 3,
ntrees = 5)
h2o.download_mojo(model, getwd(), FALSE)
실행 PrintMojo 도구 (h2o.jar 내부 포장)에 여기에 전체 문서를 참조하십시오 .png 파일을 만드는 명령 행. http://www.h2o.ai/download/에서 최신 안정 H2O3 릴리즈를 다운로드하고 명령 행에서 PrintMojo 도구를 실행해야합니다.
질문에 답이있을 경우 아래 답변을 수락하십시오. 감사! – TomKraljevic