{
‘age’: [30],
‘release_year’: [2020],
‘length_minutes’: [110]
}, index=[‘new_observation’])
Adjust age and predict
pred_dat_age = df_reviews.assign(age=obs_of_interest.loc[‘new_observation’, ‘age’])
pred_age_30 = model_lr_3feat.predict(pred_dat_age)
Adjust release year and predict
pred_dat_year = df_reviews.assign(release_year=obs_of_interest.loc[‘new_observation’, ‘release_year’])
pred_year_2020 = model_lr_3feat.predict(pred_dat_year)
Adjust length minutes and predict
pred_dat_length = df_reviews.assign(length_minutes=obs_of_interest.loc[‘new_observation’, ‘length_minutes’])
pred_length_110 = model_lr_3feat.predict(pred_dat_length)
Calculate SHAP values
shap_value_ours = pd.DataFrame({
‘age’: [mean(pred_age_30) – avg_pred],
‘release_year’: [mean(pred_year_2020) – avg_pred],
‘length_minutes’: [mean(pred_length_110) – avg_pred]
}
Leave a Reply