Skip to main content
Warning: this assignment is out of date. It may still need to be updated for this year's class. Check with your instructor before you start working on this assignment.
This assignment is due on Monday, March 21, 2022 at 11:59PM EST.
You can download the materials for this assignment here:

Homework 5: COMET-ATOMIC Schema


ATOMIC is a crowdsourced commonsense knowledge graph that is used for state-of-the-art commonsense reasoning tasks. In this homework, you will be working with COMET-ATOMIC-2020, a BART transformer model finetuned on an updated version of the original ATOMIC. You will take the information provided to you by COMET-ATOMIC-2020 and structure it into a schema used to track the state of a story.

Go to the colab notebook link above for more information.

What to Submit

  1. An IPython notebook called HW5_Schemas.ipynb that runs your COMET-ATOMIC schema. Important: Save the output for the Story Tracking Questions!
  2. A document (doc/docx, txt, pdf) that has your schema explanation and the answers to the COMET-ATOMIC questions found at the end of the notebook.

Submissions should be done on Gradescope.


  • Code - 2 points
  • Schema Diagram/Explanation - 2 points
  • Story Tracking Questions - 15 points (3 pts each)
  • COMET-ATOMIC Questions - 6 points (2 pts each)