Added scaling_type to Integer and Continuous ranges

Author: Vikram Rajasekaran

src/sagemaker/parameter.py

@@ -25,7 +25,7 @@ class ParameterRange(object):
 
     __all_types__ = ('Continuous', 'Categorical', 'Integer')
 
-    def __init__(self, min_value, max_value):
+    def __init__(self, min_value, max_value, scaling_type=None):
         """Initialize a parameter range.
 
         Args:
@@ -34,6 +34,7 @@ def __init__(self, min_value, max_value):
         """
         self.min_value = min_value
         self.max_value = max_value
+        self.scaling_type = scaling_type
 
     def is_valid(self, value):
         """Determine if a value is valid within this ParameterRange.
@@ -60,9 +61,16 @@ def as_tuning_range(self, name):
         Returns:
             dict[str, str]: A dictionary that contains the name and values of the hyperparameter.
         """
-        return {'Name': name,
-                'MinValue': to_str(self.min_value),
-                'MaxValue': to_str(self.max_value)}
+        tuning_range = {
+            'Name': name,
+            'MinValue': to_str(self.min_value),
+            'MaxValue': to_str(self.max_value)
+        }
+
+        if self.scaling_type is not None:
+            tuning_range['ScalingType'] = self.scaling_type
+
+        return tuning_range
 
 
 class ContinuousParameter(ParameterRange):

tests/unit/test_tuner.py

@@ -637,6 +637,12 @@ def test_continuous_parameter_ranges():
     assert ranges['MaxValue'] == '0.01'
 
 
+def test_continuous_parameter_scaling_type():
+    cont_param = ContinuousParameter(0.1, 2, scaling_type='ReverseLogarithmic')
+    cont_range = cont_param.as_tuning_range('range')
+    assert cont_range['ScalingType'] == 'ReverseLogarithmic'
+
+
 def test_integer_parameter():
     int_param = IntegerParameter(1, 2)
     assert isinstance(int_param, ParameterRange)
@@ -652,6 +658,12 @@ def test_integer_parameter_ranges():
     assert ranges['MaxValue'] == '2'
 
 
+def test_integer_parameter_scaling_type():
+    int_param = IntegerParameter(2, 3, scaling_type='Auto')
+    int_range = int_param.as_tuning_range('range')
+    assert int_range['ScalingType'] == 'Auto'
+
+
 def test_categorical_parameter_list():
     cat_param = CategoricalParameter(['a', 'z'])
     assert isinstance(cat_param, ParameterRange)