Skip to content
Open
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion code/pipeline/multiple_models_app.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@ def main():
# loading the data
@st.cache
def load_data():
return tm.load_data("../../annotations")
return tm.load_data("../../annotations/evaluation_agreement_1")

raw_data, dummy_data = load_data()
labels = [col for col in dummy_data if col.startswith('Class')]
Expand Down
5 changes: 5 additions & 0 deletions code/pipeline/train_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -241,6 +241,11 @@ def load_data(folder):

if not infile.is_file():
continue

# disregard the agreement files
if infile.name.startswith("agreement"):
continue

# extract the class/category name
infile_name = infile.name[infile.name.index("_"):]
infile_cat = infile_name[7:-4]
Expand Down