Bug fixes (#1891)

* fixes #1890 * fixes #1889
apple · May 20, 2019 · 97822e2 · 97822e2
1 parent 64ca3bb
commit 97822e2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/unity/python/turicreate/toolkits/_internal_utils.py b/src/unity/python/turicreate/toolkits/_internal_utils.py
@@ -721,7 +721,7 @@ def _print_neural_compute_device(cuda_gpus, use_mps, cuda_mem_req=None, has_mps_
         from ._mps_utils import mps_device_name
         print('Using GPU to create model ({})'.format(mps_device_name()))
     elif num_cuda_gpus >= 1:
-        from . import _mxnet_utils
+        from ._mxnet import _mxnet_utils
         plural = 's' if num_cuda_gpus >= 2 else ''
         print('Using GPU{} to create model ({})'.format(plural, gpu_names))
         if cuda_mem_req is not None:

diff --git a/src/unity/python/turicreate/toolkits/image_classifier/image_classifier.py b/src/unity/python/turicreate/toolkits/image_classifier/image_classifier.py
@@ -762,9 +762,10 @@ def l2Dist(v1, v2):
                 dist +=  math.pow(v1['pos'][i] - v2['pos'][i], 2)
             return math.pow(dist, 0.5)
 
-
 
-        evaluation_result = {k: metrics[k] for k in ['accuracy', 'f1_score', 'log_loss', 'precision', 'recall', 'auc']}
+
+        evaluation_result = {k: metrics[k] for k in ['accuracy', 'f1_score', 'log_loss', 'precision',
+                                                     'recall', 'auc', 'roc_curve', 'confusion_matrix']}
         evaluation_result['num_test_examples'] = len(dataset)
         for k in ['num_classes', 'num_features', 'input_image_shape', 'num_examples', 'training_loss', 'training_time', 'model', 'max_iterations']:
             evaluation_result[k] = getattr(self, k)