shared_resources/benchmark/datasets/deeper_medium/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": 10000,
    "factor_modulus": null,
    "factor_remainder": null,
    "fixed_p": null,
    "fixed_q": null,
    "fixed_time_steps": null,
    "generator_family": "rsa_repeated_squaring",
    "modulus_bits": [
      12,
      14,
      16
    ],
    "ood_examples_per_setting": 1000,
    "ood_n_depth_evaluation_examples_per_setting": 256,
    "ood_n_depth_evaluation_modulus_bits": [
      13,
      15,
      18
    ],
    "ood_time_steps": [
      16
    ],
    "output_dir": "datasets/deeper_medium",
    "seed": 1102,
    "separate_input_output": true,
    "separate_ood_splits": false,
    "split_group": "prompt",
    "test_fraction": 0.1,
    "time_steps": [
      2,
      4,
      8
    ],
    "train_fraction": 0.9
  },
  "label_format": "tail_aligned_decimal_residue",
  "label_method": "trapdoor_phi",
  "max_modulus_bits": 18,
  "max_seq_len": 17,
  "max_time_steps": 64,
  "num_examples": 103752,
  "split_counts": {
    "depth_ood_n_t_1": 768,
    "depth_ood_n_t_16": 768,
    "depth_ood_n_t_2": 768,
    "depth_ood_n_t_32": 768,
    "depth_ood_n_t_4": 768,
    "depth_ood_n_t_64": 768,
    "depth_ood_n_t_8": 768,
    "depth_t_1": 768,
    "depth_t_16": 768,
    "depth_t_2": 768,
    "depth_t_32": 768,
    "depth_t_4": 768,
    "depth_t_64": 768,
    "depth_t_8": 768,
    "ood": 3000,
    "test": 9000,
    "train": 81000
  },
  "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
}