Skip to main content
Figure 4 | EPJ Data Science

Figure 4

From: CORAL: COde RepresentAtion learning with weakly-supervised transformers for analyzing data analysis

Figure 4

Example predictions. Probability distributions over stages from CORAL’s SoftMax output (Eq. (7)) are listed on the right side. In (a), CORAL correctly identifies the cell as EVALUATE rather than WRANGLE, likely by interpreting “confusion matrix”, perhaps based on previously seen markdown. In (b), the model identifies the use of sns.regplot, an unseen statistical visualization function, as an example of EXPLORE. In (c), CORAL correctly interprets a user-defined function

Back to article page