neural-question-generator / default.json
default.json
Raw
{
  "experiment_name": "baseline",
  "dataset": {
    "training_file_path": "data/train_processed.pickle",
    "validation_file_path": "data/val_processed.pickle",
    "test_file_path": "data/test_processed.pickle",
    "vocabulary_threshold": 2,
    "batch_size": 64,
    "num_workers": 4
  },
  "experiment": {
    "num_epochs": 2,
    "learning_rate": 5e-4
  },
  "model": {
    "hidden_size": 512,
    "num_layers": 2,
    "embedding_size": 300,
    "model_type": "LSTM"
  },
  "generation": {
    "max_length": 20,
    "deterministic": false,
    "temperature": 0.1
  }
}