File tree Expand file tree Collapse file tree 1 file changed +9
-0
lines changed Expand file tree Collapse file tree 1 file changed +9
-0
lines changed Original file line number Diff line number Diff line change 11
11
# ANY KIND, either express or implied. See the License for the specific
12
12
# language governing permissions and limitations under the License.
13
13
from __future__ import absolute_import
14
+ from copy import deepcopy
14
15
15
16
import logging
16
17
import json
@@ -3825,6 +3826,12 @@ def test_script_mode_estimator_same_calls_as_framework(
3825
3826
3826
3827
model_uri = "s3://someprefix2/models/model.tar.gz"
3827
3828
training_data_uri = "s3://bucket/mydata"
3829
+ hyperparameters = {
3830
+ "int_hyperparam" : 1 ,
3831
+ "string_hyperparam" : "hello" ,
3832
+ "stringified_numeric_hyperparam" : "44" ,
3833
+ "float_hyperparam" : 1.234 ,
3834
+ }
3828
3835
3829
3836
generic_estimator = Estimator (
3830
3837
entry_point = SCRIPT_PATH ,
@@ -3838,6 +3845,7 @@ def test_script_mode_estimator_same_calls_as_framework(
3838
3845
model_uri = model_uri ,
3839
3846
dependencies = [],
3840
3847
debugger_hook_config = {},
3848
+ hyperparameters = deepcopy (hyperparameters ),
3841
3849
)
3842
3850
generic_estimator .fit (training_data_uri )
3843
3851
@@ -3858,6 +3866,7 @@ def test_script_mode_estimator_same_calls_as_framework(
3858
3866
model_uri = model_uri ,
3859
3867
dependencies = [],
3860
3868
debugger_hook_config = {},
3869
+ hyperparameters = deepcopy (hyperparameters ),
3861
3870
)
3862
3871
framework_estimator .fit (training_data_uri )
3863
3872
You can’t perform that action at this time.
0 commit comments