DEVCamiloSepulveda commited on
Commit
c9c7f56
·
verified ·
1 Parent(s): 3812e8f

Upload folder using huggingface_hub

Browse files
Files changed (5) hide show
  1. README.md +88 -0
  2. adapter_config.json +37 -0
  3. adapter_model.safetensors +3 -0
  4. pytorch_model.bin +3 -0
  5. vocab.txt +0 -0
README.md ADDED
@@ -0,0 +1,88 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: llama3.2
3
+ language:
4
+ - en
5
+ base_model: meta-llama/Llama-3.2-1B
6
+ pipeline_tag: text-classification
7
+ library_name: peft
8
+ tags:
9
+ - regression
10
+ - story-point-estimation
11
+ - software-engineering
12
+ datasets:
13
+ - talenddataquality
14
+ - appceleratorstudio
15
+ metrics:
16
+ - mae
17
+ - mdae
18
+ model-index:
19
+ - name: llama-3.2-1b-story-point-estimation
20
+ results:
21
+ - task:
22
+ type: regression
23
+ name: Story Point Estimation
24
+ dataset:
25
+ name: appceleratorstudio Dataset
26
+ type: appceleratorstudio
27
+ split: test
28
+ metrics:
29
+ - type: mae
30
+ value: 3.211
31
+ name: Mean Absolute Error (MAE)
32
+ - type: mdae
33
+ value: 3.152
34
+ name: Median Absolute Error (MdAE)
35
+ ---
36
+ # LLAMA 3 Story Point Estimator - talenddataquality - appceleratorstudio
37
+
38
+ This model is fine-tuned on issue descriptions from talenddataquality and tested on appceleratorstudio for story point estimation.
39
+
40
+ ## Model Details
41
+ - Base Model: LLAMA 3.2 1B
42
+ - Training Project: talenddataquality
43
+ - Test Project: appceleratorstudio
44
+ - Task: Story Point Estimation (Regression)
45
+ - Architecture: PEFT (LoRA)
46
+ - Tokenizer: SP WordPiece
47
+
48
+ - Input: Issue titles
49
+ - Output: Story point estimation (continuous value)
50
+
51
+ ## Usage
52
+ ```python
53
+ from transformers import AutoModelForSequenceClassification, BertTokenizer
54
+ from peft import PeftConfig, PeftModel
55
+
56
+ # Load peft config model
57
+ config = PeftConfig.from_pretrained("DEVCamiloSepulveda/666-LLAMA3SP-talenddataquality-appceleratorstudio")
58
+
59
+ # Load tokenizer and model
60
+ tokenizer = BertTokenizer('vocab.txt')
61
+ base_model = AutoModelForSequenceClassification.from_pretrained(
62
+ config.base_model_name_or_path,
63
+ num_labels=1,
64
+ torch_dtype=torch.float16,
65
+ device_map='auto'
66
+ )
67
+ model = PeftModel.from_pretrained(base_model, "DEVCamiloSepulveda/666-LLAMA3SP-talenddataquality-appceleratorstudio")
68
+
69
+ # Prepare input text
70
+ text = "Your issue description here"
71
+ inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=20, padding="max_length")
72
+
73
+ # Get prediction
74
+ outputs = model(**inputs)
75
+ story_points = outputs.logits.item()
76
+ ```
77
+
78
+ ## Training Details
79
+ - Fine-tuning method: LoRA (Low-Rank Adaptation)
80
+ - Sequence length: 20 tokens
81
+ - Best training epoch: 6 / 20 epochs
82
+ - Batch size: 32
83
+ - Training time: 484.246 seconds
84
+ - Mean Absolute Error (MAE): 3.211
85
+ - Median Absolute Error (MdAE): 3.152
86
+ ### Framework versions
87
+
88
+ - PEFT 0.14.0
adapter_config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "meta-llama/Llama-3.2-1B",
5
+ "bias": "none",
6
+ "eva_config": null,
7
+ "exclude_modules": null,
8
+ "fan_in_fan_out": false,
9
+ "inference_mode": true,
10
+ "init_lora_weights": true,
11
+ "layer_replication": null,
12
+ "layers_pattern": null,
13
+ "layers_to_transform": null,
14
+ "loftq_config": {},
15
+ "lora_alpha": 16,
16
+ "lora_bias": false,
17
+ "lora_dropout": 0.1,
18
+ "megatron_config": null,
19
+ "megatron_core": "megatron.core",
20
+ "modules_to_save": [
21
+ "classifier",
22
+ "score"
23
+ ],
24
+ "peft_type": "LORA",
25
+ "r": 8,
26
+ "rank_pattern": {},
27
+ "revision": null,
28
+ "target_modules": [
29
+ "q_proj",
30
+ "o_proj",
31
+ "k_proj",
32
+ "v_proj"
33
+ ],
34
+ "task_type": "SEQ_CLS",
35
+ "use_dora": false,
36
+ "use_rslora": false
37
+ }
adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e10c07641a02af10de588627a97219ddf2caade4e93f834485ddeeb0f6534fd
3
+ size 6840816
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c02d56c991c7999d6a3229068ea76b7d8da4fa663b8913f3152ae08ae74d048
3
+ size 1560270490
vocab.txt ADDED
The diff for this file is too large to render. See raw diff