bjorn-hommel's picture
minor fixes
021a052
raw
history blame
1.23 kB
import streamlit as st
import pandas as pd
covariate_columns = {
'content_domain': 'Content Domain',
'language': 'Language',
'rater_group': 'Rater Group',
}
id_vars = [
'mean_z', 'text', 'content_domain', 'language',
'rater_group', 'study', 'instrument'
]
def load_data():
st.session_state.df = (
pd
.read_feather(path='data.feather')
.query('partition == "test" | partition == "dev"')
.melt(
value_vars=['sentiment_model', 'desirability_model'],
var_name='x_group',
value_name='x',
id_vars=id_vars
)
.replace(
to_replace={
'en': 'English',
'de': 'German',
'other': 'Other',
'personality': 'Personality',
'laypeople': 'Laypeople',
'students': 'Students',
'sentiment_model': 'Sentiment Model',
'desirability_model': 'Desirability Model'
}
)
.rename(columns=covariate_columns)
.rename(
columns={
'mean_z': 'Human-ratings',
'x': 'Machine-ratings',
}
)
)