Goal inference

# goal_prob ~ true_state + utterance + goal
ms <- d %>%
  group_by(goal, true_state, positivity, utterance, subid) %>%
  summarize(
            goal_prob = mean(goal_prob, na.rm=TRUE)
          ) %>%
  group_by(goal, true_state, positivity, utterance) %>%
  multi_boot_standard(column = "goal_prob") %>%
  mutate(goal_prob = mean)

## Joining, by = c("goal", "true_state", "positivity", "utterance")

# ms$goal <- ordered(ms$goal, levels = c("honest", "nice", "mean"))
ms$positivity <- factor(ms$positivity, levels = c("positive", "negative"))
ms$positivity <- factor(ms$positivity, labels = c("direct (no negation)", "indirect (negation)"))
levels(ms$true_state) <- c("1 heart", "2 hearts", "3 hearts", "4 hearts", "5 hearts")

qplot(utterance, goal_prob, 
      colour = goal,
      linetype = positivity,
      data=ms) + 
  geom_line(aes(group=interaction(goal, positivity), col=goal)) +
  facet_grid(.~true_state) +
  xlab("words") +
  ylab("likelihood for goal") +
 # ggtitle("Experiment: goals given heart-state and utterance") +
  geom_errorbar(aes(ymin=ci_lower,ymax=ci_upper,width=.1)) +
  geom_hline(yintercept=.5, lty=2) +
  scale_linetype_discrete(guide = guide_legend(title = "utterance type"))

qplot(utterance, goal_prob, 
      colour = goal,
      data=ms) + 
  geom_line(aes(group=goal, col=goal)) +
  facet_grid(positivity~true_state) +
  xlab("words") +
  ylab("likelihood for goal") +
 # ggtitle("Experiment: goals given heart-state and utterance") +
  geom_errorbar(aes(ymin=ci_lower,ymax=ci_upper,width=.1)) +
  geom_hline(yintercept=0.5, lty=2) +
  scale_linetype_discrete(guide = guide_legend(title = "utterance type")) +
  theme(axis.text.x = element_text(angle = 45, hjust = 1))

ms <- d %>%
  group_by(goal, true_state, positivity, subid) %>%
  summarize(
            goal_prob = mean(goal_prob, na.rm=TRUE)
          ) %>%
  group_by(goal, true_state, positivity) %>%
  multi_boot_standard(column = "goal_prob") %>%
  mutate(goal_prob = mean)

## Joining, by = c("goal", "true_state", "positivity")

# ms$goal <- ordered(ms$goal, levels = c("honest", "nice", "mean"))
ms$positivity <- factor(ms$positivity, levels = c("positive", "negative"))
ms$positivity <- factor(ms$positivity, labels = c("direct (no negation)", "indirect (negation)"))
levels(ms$true_state) <- c("1 heart", "2 hearts", "3 hearts", "4 hearts", "5 hearts")

qplot(true_state, goal_prob, 
      colour = goal,
      data=ms) + 
  geom_line(aes(group=goal, col=goal)) +
  facet_grid(.~positivity) +
  xlab("words") +
  ylab("likelihood for goal") +
 # ggtitle("Experiment: goals given heart-state and utterance") +
  geom_errorbar(aes(ymin=ci_lower,ymax=ci_upper,width=.1)) +
  geom_hline(yintercept=.5, lty=2) +
  scale_linetype_discrete(guide = guide_legend(title = "utterance type"))

```

head(d)

##    subid   goal true_state positivity utterance goal_prob
## 1 anon-0 social          3   negative  terrible     0.000
## 2 anon-0 social          5   positive   amazing     0.875
## 3 anon-0 social          2   positive       bad     0.025
## 4 anon-0 social          4   negative       bad     0.150
## 5 anon-0 social          2   positive      okay     0.075
## 6 anon-0 social          3   positive      good     0.600

ms <- d %>%
  filter(!is.na(goal_prob)) %>%
  mutate(goal_prob = ifelse(goal_prob == 0, 0.01, goal_prob)) %>% # to avoid 0 probabilities
  mutate(entropy = goal_prob*log(1/goal_prob)) %>%
  group_by(true_state, positivity, utterance) %>%
  multi_boot(summary_function = "sum", column = "entropy", summary_groups =c("true_state", "positivity", "utterance"),
           statistics_functions = c("ci_lower", "mean", "ci_upper"))

ms$positivity <- factor(ms$positivity, levels = c("positive", "negative"))
ms$positivity <- factor(ms$positivity, labels = c("direct (no negation)", "indirect (negation)"))
levels(ms$true_state) <- c("1 heart", "2 hearts", "3 hearts", "4 hearts", "5 hearts")

qplot(utterance, mean, 
      colour = positivity,
      data=ms) + 
  geom_line(aes(group=positivity, col=positivity)) +
  facet_grid(.~true_state) +
  xlab("words") +
  ylab("entropy") +
 # ggtitle("Experiment: goals given heart-state and utterance") +
  geom_errorbar(aes(ymin=ci_lower,ymax=ci_upper,width=.1)) +
  # geom_hline(yintercept=0.5, lty=2) +
  scale_colour_brewer(guide = guide_legend(title = "utterance type"), palette = "Set1") +
  theme(axis.text.x = element_text(angle = 45, hjust = 1))