Team Ai
Modelpublic

Lifehouse/distilbert-sql-timeout-classifier-with-features-4096-sql-normalized

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face
0likes7downloads
README.md80 linesDownload Raw Back to root
1---2license: apache-2.03base_model: distilbert-base-uncased4tags:5- generated_from_trainer6datasets:7- generator8metrics:9- accuracy10model-index:11- name: distilbert-sql-timeout-classifier-with-features-4096-sql-normalized12  results:13  - task:14      name: Text Classification15      type: text-classification16    dataset:17      name: generator18      type: generator19      config: default20      split: train21      args: default22    metrics:23    - name: Accuracy24      type: accuracy25      value: 0.890603319087528426---27 28<!-- This model card has been generated automatically according to the information the Trainer had access to. You29should probably proofread and complete it, then remove this comment. -->30 31# distilbert-sql-timeout-classifier-with-features-4096-sql-normalized32 33This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the generator dataset.34It achieves the following results on the evaluation set:35- Loss: 0.559836- Accuracy: 0.890637 38## Model description39 40More information needed41 42## Intended uses & limitations43 44More information needed45 46## Training and evaluation data47 48More information needed49 50## Training procedure51 52### Training hyperparameters53 54The following hyperparameters were used during training:55- learning_rate: 2e-0556- train_batch_size: 457- eval_batch_size: 458- seed: 4259- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-0860- lr_scheduler_type: linear61- num_epochs: 562 63### Training results64 65| Training Loss | Epoch | Step | Validation Loss | Accuracy |66|:-------------:|:-----:|:----:|:---------------:|:--------:|67| 0.5057        | 1.0   | 1938 | 0.4010          | 0.8793   |68| 0.3304        | 2.0   | 3876 | 0.4271          | 0.8945   |69| 0.2143        | 3.0   | 5814 | 0.4978          | 0.8872   |70| 0.2079        | 4.0   | 7752 | 0.6021          | 0.8776   |71| 0.1329        | 5.0   | 9690 | 0.5598          | 0.8906   |72 73 74### Framework versions75 76- Transformers 4.38.177- Pytorch 2.2.1+cu12178- Datasets 2.17.179- Tokenizers 0.15.280