JRJ/model-registry
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Small fix

Browse Source
This commit is contained in:
Jason Jafari 2024-06-16 16:58:17 -04:00
parent 09da3b09f9
commit 7f8fdfa05e
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mlModelSaver/__init__.py
View File

@ -77,7 +77,6 @@ def mlModelSavePredict(self, df, typeOfPredict = 'normal'):
modelTypeConfig = supportedModels[modelType] modelTypeConfig = supportedModels[modelType]
if typeOfPredict == 'normal': if typeOfPredict == 'normal':
results = getattr(self, modelTypeConfig['normalPredictorFunction'])(dfAfterTransformation) results = getattr(self, modelTypeConfig['normalPredictorFunction'])(dfAfterTransformation)
print(results)
for value in results: for value in results:
if isinstance(value, np.ndarray): if isinstance(value, np.ndarray):
res = {} res = {}
@ -88,9 +87,6 @@ def mlModelSavePredict(self, df, typeOfPredict = 'normal'):
output.append({ output.append({
outputsName[0]: value, outputsName[0]: value,
}) })
print(type(value))
print(value)
print(outputsName)
return output return output
return output return output