{
  "source": [
    "red",
    "square"
  ],
  "reference": [
    "carré",
    "rouge",
    "<EOS>"
  ],
  "teacher": {
    "trace": [
      {
        "step": 1,
        "prefix": [],
        "input": "<BOS>",
        "distribution": {
          "carré": 0.4,
          "cercle": 0.6
        },
        "top": "cercle",
        "chosen": "carré",
        "chosen_probability": 0.4,
        "sequence_probability": 0.4
      },
      {
        "step": 2,
        "prefix": [
          "carré"
        ],
        "input": "carré",
        "distribution": {
          "rouge": 0.9,
          "bleu": 0.1
        },
        "top": "rouge",
        "chosen": "rouge",
        "chosen_probability": 0.9,
        "sequence_probability": 0.36000000000000004
      },
      {
        "step": 3,
        "prefix": [
          "carré",
          "rouge"
        ],
        "input": "rouge",
        "distribution": {
          "<EOS>": 1.0
        },
        "top": "<EOS>",
        "chosen": "<EOS>",
        "chosen_probability": 1.0,
        "sequence_probability": 0.36000000000000004
      }
    ],
    "tokens": [
      "carré",
      "rouge",
      "<EOS>"
    ],
    "probability": 0.36000000000000004
  },
  "greedy": {
    "trace": [
      {
        "step": 1,
        "prefix": [],
        "input": "<BOS>",
        "distribution": {
          "carré": 0.4,
          "cercle": 0.6
        },
        "top": "cercle",
        "chosen": "cercle",
        "chosen_probability": 0.6,
        "sequence_probability": 0.6
      },
      {
        "step": 2,
        "prefix": [
          "cercle"
        ],
        "input": "cercle",
        "distribution": {
          "rouge": 0.5,
          "bleu": 0.5
        },
        "top": "rouge",
        "chosen": "rouge",
        "chosen_probability": 0.5,
        "sequence_probability": 0.3
      },
      {
        "step": 3,
        "prefix": [
          "cercle",
          "rouge"
        ],
        "input": "rouge",
        "distribution": {
          "<EOS>": 1.0
        },
        "top": "<EOS>",
        "chosen": "<EOS>",
        "chosen_probability": 1.0,
        "sequence_probability": 0.3
      }
    ],
    "tokens": [
      "cercle",
      "rouge",
      "<EOS>"
    ],
    "probability": 0.3
  },
  "beams": {
    "1": {
      "width": 1,
      "history": [
        {
          "step": 1,
          "candidates": [
            {
              "tokens": [
                "cercle"
              ],
              "probability": 0.6,
              "retained": true
            },
            {
              "tokens": [
                "carré"
              ],
              "probability": 0.4,
              "retained": false
            }
          ]
        },
        {
          "step": 2,
          "candidates": [
            {
              "tokens": [
                "cercle",
                "rouge"
              ],
              "probability": 0.3,
              "retained": true
            },
            {
              "tokens": [
                "cercle",
                "bleu"
              ],
              "probability": 0.3,
              "retained": false
            }
          ]
        },
        {
          "step": 3,
          "candidates": [
            {
              "tokens": [
                "cercle",
                "rouge",
                "<EOS>"
              ],
              "probability": 0.3,
              "retained": true
            }
          ]
        }
      ],
      "tokens": [
        "cercle",
        "rouge",
        "<EOS>"
      ],
      "probability": 0.3
    },
    "2": {
      "width": 2,
      "history": [
        {
          "step": 1,
          "candidates": [
            {
              "tokens": [
                "cercle"
              ],
              "probability": 0.6,
              "retained": true
            },
            {
              "tokens": [
                "carré"
              ],
              "probability": 0.4,
              "retained": true
            }
          ]
        },
        {
          "step": 2,
          "candidates": [
            {
              "tokens": [
                "carré",
                "rouge"
              ],
              "probability": 0.36000000000000004,
              "retained": true
            },
            {
              "tokens": [
                "cercle",
                "rouge"
              ],
              "probability": 0.3,
              "retained": true
            },
            {
              "tokens": [
                "cercle",
                "bleu"
              ],
              "probability": 0.3,
              "retained": false
            },
            {
              "tokens": [
                "carré",
                "bleu"
              ],
              "probability": 0.04000000000000001,
              "retained": false
            }
          ]
        },
        {
          "step": 3,
          "candidates": [
            {
              "tokens": [
                "carré",
                "rouge",
                "<EOS>"
              ],
              "probability": 0.36000000000000004,
              "retained": true
            },
            {
              "tokens": [
                "cercle",
                "rouge",
                "<EOS>"
              ],
              "probability": 0.3,
              "retained": true
            }
          ]
        }
      ],
      "tokens": [
        "carré",
        "rouge",
        "<EOS>"
      ],
      "probability": 0.36000000000000004
    }
  },
  "reference_nll": 1.0216512475319812,
  "lags": {
    "forward": [
      4,
      4,
      4,
      4
    ],
    "reversed": [
      1,
      3,
      5,
      7
    ]
  }
}