shared_resources/benchmark/datasets/deeper_easy/dataset_config.json

{
  "data_format": "separate_input_output",
  "dataset_kind": "squaring_mod",
  "factor_congruence": null,
  "generator_config": {
    "depth_evaluation_examples_per_setting": 256,
    "depth_evaluation_exhaustive_x": false,
    "depth_evaluation_time_steps": [
      1,
      2,
      4,
      8,
      16,
      32,
      64
    ],
    "examples_per_setting": 1000,
    "factor_modulus": null,
    "factor_remainder": null,
    "fixed_p": null,
    "fixed_q": null,
    "fixed_time_steps": null,
    "generator_family": "rsa_repeated_squaring",
    "modulus_bits": [
      10,
      11
    ],
    "ood_examples_per_setting": 300,
    "ood_n_depth_evaluation_examples_per_setting": 256,
    "ood_n_depth_evaluation_modulus_bits": [
      12,
      13
    ],
    "ood_time_steps": [
      6
    ],
    "output_dir": "datasets/deeper_easy",
    "seed": 1101,
    "separate_input_output": true,
    "separate_ood_splits": false,
    "split_group": "prompt",
    "test_fraction": 0.2,
    "time_steps": [
      1,
      2,
      3
    ],
    "train_fraction": 0.8
  },
  "label_format": "tail_aligned_decimal_residue",
  "label_method": "trapdoor_phi",
  "max_modulus_bits": 13,
  "max_seq_len": 13,
  "max_time_steps": 64,
  "num_examples": 13768,
  "split_counts": {
    "depth_ood_n_t_1": 512,
    "depth_ood_n_t_16": 512,
    "depth_ood_n_t_2": 512,
    "depth_ood_n_t_32": 512,
    "depth_ood_n_t_4": 512,
    "depth_ood_n_t_64": 512,
    "depth_ood_n_t_8": 512,
    "depth_t_1": 512,
    "depth_t_16": 512,
    "depth_t_2": 512,
    "depth_t_32": 512,
    "depth_t_4": 512,
    "depth_t_64": 512,
    "depth_t_8": 512,
    "ood": 600,
    "test": 1200,
    "train": 4800
  },
  "split_group": "prompt",
  "token_ids": {
    "ANS": 5,
    "BOS": 1,
    "DIGIT_OFFSET": 7,
    "EOS": 6,
    "N": 2,
    "PAD": 0,
    "T": 4,
    "X": 3
  },
  "vocab_size": 17
}