4
4
import json
5
5
6
6
from .._services .model_repository import ModelRepository as mr
7
+
7
8
# TODO: Convert STRINGIO calls to string or dict format
8
9
9
10
@@ -31,7 +32,9 @@ def _find_file(model, file_name):
31
32
for file in file_list :
32
33
print (file .name )
33
34
if file_name .lower () in file .name .lower ():
34
- correct_file = sess .get (f"modelRepository/models/{ model } /contents/{ file .id } /content" )
35
+ correct_file = sess .get (
36
+ f"modelRepository/models/{ model } /contents/{ file .id } /content"
37
+ )
35
38
return correct_file
36
39
37
40
@@ -57,13 +60,16 @@ def generate_hyperparameters(cls, model, model_prefix, pickle_path):
57
60
JSON file
58
61
Named {model_prefix}Hyperparameters.json.
59
62
"""
63
+
60
64
def sklearn_params ():
61
65
"""
62
66
Generates hyperparameters for the models generated by scikit-learn.
63
67
"""
64
68
hyperparameters = model .get_params ()
65
69
model_json = {"hyperparameters" : hyperparameters }
66
- with open (Path (pickle_path ) / f"{ model_prefix } Hyperparameters.json" , "w" ) as f :
70
+ with open (
71
+ Path (pickle_path ) / f"{ model_prefix } Hyperparameters.json" , "w"
72
+ ) as f :
67
73
f .write (json .dumps (model_json , indent = 4 ))
68
74
69
75
if all (hasattr (model , attr ) for attr in ["_estimator_type" , "get_params" ]):
@@ -162,5 +168,5 @@ def add_hyperparameters(cls, model, **kwargs):
162
168
mr .add_model_content (
163
169
model ,
164
170
StringIO (json .dumps (hyperparameters , indent = 4 )),
165
- f"{ model .name } Hyperparameters.json"
171
+ f"{ model .name } Hyperparameters.json" ,
166
172
)
0 commit comments