paperswithcode_nbow / config.py
lambdaofgod's picture
additional cols and optional device
a284f57
raw
history blame
358 Bytes
model_names = [
"dependencies-nbow-nbow-mnrl",
"readme-nbow-nbow-mnrl",
"titles-nbow-nbow-mnrl",
"titles_dependencies-nbow-nbow-mnrl",
"readme_dependencies-nbow-nbow-mnrl",
]
best_tasks_path = "assets/best_tasks.csv"
worst_tasks_path = "assets/worst_tasks.csv"
text_cols = ["dependencies", "readme", "titles"]
text_list_cols = ["titles"]