From 2048db33fc316072351d88d3ff44ff6f4fe37959 Mon Sep 17 00:00:00 2001 From: Andy Eschbacher Date: Thu, 22 Mar 2018 13:12:17 -0400 Subject: [PATCH] avoids accuracy calculation without model being defined --- src/py/crankshaft/crankshaft/segmentation/segmentation.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/py/crankshaft/crankshaft/segmentation/segmentation.py b/src/py/crankshaft/crankshaft/segmentation/segmentation.py index c78c9b3..8137a0a 100644 --- a/src/py/crankshaft/crankshaft/segmentation/segmentation.py +++ b/src/py/crankshaft/crankshaft/segmentation/segmentation.py @@ -80,6 +80,11 @@ class Segmentation(object): # find model if it exists and is specified if model_name is not None: model = model_storage.get_model(model_name) + _, features_test, _, target_test = train_test_split( + features, target, test_size=0.2) + if model is not None: + accuracy = calculate_model_accuracy( + model, features_test, target_test) if locals().get('model') is None: model, accuracy = train_model(target, features, model_params, 0.2)