config.json

{
  "model_type": "vae",
  "architectures": ["VariationalAutoencoder"],
  "model_name": "VAE-Banking77-OpenIntentClassification",
  
  "input_dim": 768,
  "hidden_dim": 256,
  "latent_dim": 64,
  
  "encoder_config": {
    "layers": [
      {"type": "Linear", "in_features": 768, "out_features": 256},
      {"type": "ReLU"}
    ],
    "mu_layer": {"type": "Linear", "in_features": 256, "out_features": 64},
    "logvar_layer": {"type": "Linear", "in_features": 256, "out_features": 64}
  },
  
  "decoder_config": {
    "layers": [
      {"type": "Linear", "in_features": 64, "out_features": 256},
      {"type": "ReLU"},
      {"type": "Linear", "in_features": 256, "out_features": 768}
    ]
  },
  
  "training_config": {
    "dataset": "PolyAI/banking77",
    "loss_weights": {
      "reconstruction_loss": 0.5,
      "kl_divergence": 0.5
    },
    "validation_metric": "reconstruction_loss"
  },
  
  "library_name": "pytorch",
  "torch_dtype": "float32",
  "transformers_version": null,
  "created_date": "2025-08-11"
}