www.flow.packs.test-small.gbm_testng_insurance_validation1.flow Maven / Gradle / Ivy
The newest version!
{
"version": "1.0.0",
"cells": [
{
"type": "cs",
"input": "importFiles"
},
{
"type": "cs",
"input": "importFiles [ \"../smalldata/testng/insurance_validation1.csv\" ]"
},
{
"type": "cs",
"input": "setupParse paths: [ \"../smalldata/testng/insurance_validation1.csv\" ]"
},
{
"type": "cs",
"input": "parseFiles\n paths: [\"../smalldata/testng/insurance_validation1.csv\"]\n destination_frame: \"insurance_validation1.hex\"\n parse_type: \"CSV\"\n separator: 44\n number_columns: 5\n single_quotes: false\n column_names: [\"District\",\"Group\",\"Age\",\"Holders\",\"Claims\"]\n column_types: [\"Numeric\",\"Enum\",\"Enum\",\"Numeric\",\"Numeric\"]\n delete_on_done: true\n check_header: 1\n chunk_size: 4194304"
},
{
"type": "cs",
"input": "getFrameSummary \"insurance_validation1.hex\""
},
{
"type": "cs",
"input": "assist splitFrame, \"insurance_validation1.hex\""
},
{
"type": "cs",
"input": "splitFrame \"insurance_validation1.hex\", [0.2], [\"insurance_validation1_test\",\"insurance_validation1_train\"], 123456"
},
{
"type": "cs",
"input": "buildModel \"gbm\""
},
{
"type": "cs",
"input": "buildModel 'gbm', {\"model_id\":\"gbm-9de3090e-9e53-4a00-ab89-f9cc1646ddfd\",\"training_frame\":\"insurance_validation1_train\",\"validation_frame\":\"insurance_validation1_test\",\"ignore_const_cols\":true,\"response_column\":\"District\",\"ntrees\":\"20\",\"max_depth\":5,\"min_rows\":\"5\",\"nbins\":20,\"nbins_cats\":1024,\"seed\":8460237478751180000,\"learn_rate\":\"0.3\",\"distribution\":\"AUTO\",\"score_each_iteration\":false,\"r2_stopping\":0.999999,\"build_tree_one_node\":false}"
},
{
"type": "cs",
"input": "getModel \"gbm-9de3090e-9e53-4a00-ab89-f9cc1646ddfd\""
},
{
"type": "cs",
"input": "predict model: \"gbm-9de3090e-9e53-4a00-ab89-f9cc1646ddfd\""
},
{
"type": "cs",
"input": "predict model: \"gbm-9de3090e-9e53-4a00-ab89-f9cc1646ddfd\", frame: \"insurance_validation1_test\", predictions_frame: \"prediction-e4b9f171-be0d-4084-a6b2-8b1d3392b24e\""
}
]
}