www.flow.packs.test-small.gbm_testng_insurance_train1.flow Maven / Gradle / Ivy
The newest version!
{
"version": "1.0.0",
"cells": [
{
"type": "cs",
"input": "importFiles"
},
{
"type": "cs",
"input": "importFiles [ \"../smalldata/testng/insurance_train1.csv\" ]"
},
{
"type": "cs",
"input": "setupParse paths: [ \"../smalldata/testng/insurance_train1.csv\" ]"
},
{
"type": "cs",
"input": "parseFiles\n paths: [\"../smalldata/testng/insurance_train1.csv\"]\n destination_frame: \"insurance_train1.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_train1.hex\""
},
{
"type": "cs",
"input": "assist splitFrame, \"insurance_train1.hex\""
},
{
"type": "cs",
"input": "splitFrame \"insurance_train1.hex\", [0.2], [\"insurance_train1_test\",\"insurance_train1_train\"], 123456"
},
{
"type": "cs",
"input": "buildModel \"gbm\""
},
{
"type": "cs",
"input": "buildModel 'gbm', {\"model_id\":\"gbm-da6388cd-a3b9-4141-84ad-6ace1c5b70a4\",\"training_frame\":\"insurance_train1_train\",\"validation_frame\":\"insurance_train1_test\",\"ignore_const_cols\":true,\"response_column\":\"District\",\"ntrees\":\"20\",\"max_depth\":5,\"min_rows\":\"10\",\"nbins\":20,\"nbins_cats\":1024,\"seed\":-5574115868883224000,\"learn_rate\":\"0.3\",\"distribution\":\"AUTO\",\"score_each_iteration\":false,\"r2_stopping\":0.999999,\"build_tree_one_node\":false}"
},
{
"type": "cs",
"input": "getModel \"gbm-da6388cd-a3b9-4141-84ad-6ace1c5b70a4\""
},
{
"type": "cs",
"input": "predict model: \"gbm-da6388cd-a3b9-4141-84ad-6ace1c5b70a4\""
},
{
"type": "cs",
"input": "predict model: \"gbm-da6388cd-a3b9-4141-84ad-6ace1c5b70a4\", frame: \"insurance_train1_test\", predictions_frame: \"prediction-77f8985b-fdb9-419f-8741-d84d16159c69\""
}
]
}