Closed cyberholics closed 1 year ago
Hi @cyberholics, I think there is a problem with your definition of prediction_function
. This function takes a dataframe as input (in your case it is a dataframe with a single column SMS
). You need to take this into account when applying your vectorizer:
def prediction_function(df):
preprocessed_df = vectorizer.transform(df["SMS"]) # note that we select the "SMS" feature here
return nb_model.predict_proba(preprocessed_df)
Thanks @mattbit!! it worked i can continue with my project
Issue Type
Bug
Source
source
Giskard Library Version
2.0.0b14
Giskard Server Version
2.0.0b11
OS Platform and Distribution
mac os 10.13.6
Python version
No response
Installed python packages
Current Behaviour?
Standalone code OR list down the steps to reproduce the issue
Relevant log output