Team Ai
Modelpublic

hiwensen/bert_sql_classfication

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face
0likes8downloads
config.json42 linesDownload Raw Back to root
1{2  "_name_or_path": "distilbert-base-uncased",3  "activation": "gelu",4  "architectures": [5    "DistilBertForSequenceClassification"6  ],7  "attention_dropout": 0.1,8  "dim": 768,9  "dropout": 0.1,10  "hidden_dim": 3072,11  "id2label": {12    "0": "where",13    "1": "table_join",14    "2": "ratio",15    "3": "order_by",16    "4": "group_by",17    "5": "date_functions"18  },19  "initializer_range": 0.02,20  "label2id": {21    "date_functions": 5,22    "group_by": 4,23    "order_by": 3,24    "ratio": 2,25    "table_join": 1,26    "where": 027  },28  "max_position_embeddings": 512,29  "model_type": "distilbert",30  "n_heads": 12,31  "n_layers": 6,32  "pad_token_id": 0,33  "problem_type": "single_label_classification",34  "qa_dropout": 0.1,35  "seq_classif_dropout": 0.2,36  "sinusoidal_pos_embds": false,37  "tie_weights_": true,38  "torch_dtype": "float32",39  "transformers_version": "4.36.2",40  "vocab_size": 3052241}42