SIENTIAPDE-1712
Add debug logging for non-DataFrame prediction data in MLFlowRepository
This commit is contained in:
@@ -1283,6 +1283,10 @@ class MLFlowRepository(SientiaMonitoring):
|
||||
predict_data.columns = pd.Index(['prediction'])
|
||||
|
||||
else:
|
||||
self.debug(
|
||||
f'Data received from model prediction (not a DataFrame): {predict_data}',
|
||||
metadata,
|
||||
)
|
||||
predict_data = pd.DataFrame(predict_data, columns=['prediction'])
|
||||
# predict_data.to_csv(
|
||||
# f"tmp/predicted_data_{model_name}.csv", index=True)
|
||||
|
||||
Reference in New Issue
Block a user