Upload folder using huggingface_hub

README.md

@@ -1,188 +1,207 @@
 ---
-license: mit
-language:
-- zh
-- en
-datasets:
-- IowaCat/page_type_inference_dataset
-metrics:
-- accuracy: 0.99
+base_model: Qwen/Qwen2.5-1.5B
+library_name: peft
 pipeline_tag: text-generation
 tags:
-- url-classification
-- list-page-detection
-- detail-page-detection
-- qwen
-- fine-tuning
+- base_model:adapter:Qwen/Qwen2.5-1.5B
 - lora
-- url-parser
-widget:
-- text: "https://example.com/product/12345"
-- text: "https://example.com/category/electronics"
+- transformers
 ---
 
-# URL Page Type Classifier
+# Model Card for Model ID
 
-<div align="center">
+<!-- Provide a quick summary of what the model is/does. -->
 
-![Model Size](https://img.shields.io/badge/Model%20Size-1.5B-blue)
-![License](https://img.shields.io/badge/License-MIT-green)
-![Accuracy](https://img.shields.io/badge/Accuracy-99%25-green)
 
-</div>
 
-## 📋 概述
+## Model Details
 
-基于 Qwen2.5-1.5B + LoRA 微调的URL类型分类模型，用于判断URL是列表页还是详情页。
+### Model Description
 
-## 🏗️ 模型架构
+<!-- Provide a longer summary of what this model is. -->
 
-| 项目 | 详情 |
-|------|------|
-| **基础模型** | Qwen/Qwen2.5-1.5B |
-| **微调方法** | LoRA (r=16, alpha=32) |
-| **参数量** | 1.5B |
-| **可训练参数** | ~18M (1.18%) |
 
-## 📊 训练数据
 
-- **数据集**: IowaCat/page_type_inference_dataset
-- **训练样本**: 10,000条URL (5000列表页 + 5000详情页)
-- **数据来源**: HuggingFace Datasets
+- **Developed by:** [More Information Needed]
+- **Funded by [optional]:** [More Information Needed]
+- **Shared by [optional]:** [More Information Needed]
+- **Model type:** [More Information Needed]
+- **Language(s) (NLP):** [More Information Needed]
+- **License:** [More Information Needed]
+- **Finetuned from model [optional]:** [More Information Needed]
 
-### 数据分布
+### Model Sources [optional]
 
-| 类型 | 数量 | 比例 |
-|------|------|------|
-| 列表页 (List Page) | 5,000 | 50% |
-| 详情页 (Detail Page) | 5,000 | 50% |
+<!-- Provide the basic links for the model. -->
 
-## ⚙️ 训练配置
+- **Repository:** [More Information Needed]
+- **Paper [optional]:** [More Information Needed]
+- **Demo [optional]:** [More Information Needed]
 
-```python
-{
-    "base_model": "Qwen/Qwen2.5-1.5B",
-    "lora_rank": 16,
-    "lora_alpha": 32,
-    "lora_dropout": 0.05,
-    "num_train_epochs": 3,
-    "per_device_train_batch_size": 2,
-    "gradient_accumulation_steps": 8,
-    "learning_rate": 2e-4,
-    "fp16": true,
-    "optimizer": "adamw_torch",
-    "lr_scheduler_type": "cosine"
-}
-```
+## Uses
 
-## 📈 性能评估
+<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
 
-### 测试结果
+### Direct Use
 
-| 测试集 | 样本数 | 准确率 |
-|--------|--------|--------|
-| 验证集 | 100 | **99%** |
+<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
 
-### 示例预测
+[More Information Needed]
 
-| URL | 预测结果 |
-|-----|----------|
-| `https://example.com/products/category` | 列表页 (List Page) |
-| `https://example.com/product/12345` | 详情页 (Detail Page) |
-| `https://example.com/search?q=test` | 列表页 (List Page) |
-| `https://example.com/item/abc123` | 详情页 (Detail Page) |
-| `https://example.com/list/all` | 列表页 (List Page) |
+### Downstream Use [optional]
 
-## 🚀 快速开始
+<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
 
-### 安装依赖
+[More Information Needed]
 
-```bash
-pip install transformers peft torch
-```
+### Out-of-Scope Use
 
-### 推理代码
+<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
 
-```python
-from transformers import AutoTokenizer, AutoModelForCausalLM
+[More Information Needed]
 
-model_name = "windlx/url-classifier-model"
-tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
+## Bias, Risks, and Limitations
 
-# 要分类的URL
-url = "https://example.com/product/12345"
+<!-- This section is meant to convey both technical and sociotechnical limitations. -->
 
-# 构建提示
-prompt = f"""请判断以下URL是列表页还是详情页。
+[More Information Needed]
 
-URL: {url}
-类型: """
+### Recommendations
 
-# 推理
-inputs = tokenizer(prompt, return_tensors="pt")
-outputs = model.generate(**inputs, max_new_tokens=10, do_sample=False)
-response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
 
-# 提取结果
-if "详情页" in response or "Detail Page" in response:
-    result = "详情页 (Detail Page)"
-else:
-    result = "列表页 (List Page)"
+Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
 
-print(f"URL: {url}")
-print(f"类型: {result}")
-```
+## How to Get Started with the Model
 
-### 使用 GPU
+Use the code below to get started with the model.
 
-```python
-# 自动使用GPU
-model = AutoModelForCausalLM.from_pretrained(
-    model_name, 
-    trust_remote_code=True,
-    device_map="auto",
-    torch_dtype="auto"
-)
-```
+[More Information Needed]
 
-### 使用 CPU
+## Training Details
 
-```python
-# 强制使用CPU
-model = AutoModelForCausalLM.from_pretrained(
-    model_name, 
-    trust_remote_code=True,
-    device_map="cpu",
-    torch_dtype="float32"
-)
-```
+### Training Data
 
-## ⚠️ 局限性
+<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
 
-1. **仅基于URL字符串** - 不访问实际网页内容
-2. **依赖URL路径规范** - 对于URL路径不规范的网站，准确率可能较低
-3. **仅支持中英文** - 主要针对中文URL优化
+[More Information Needed]
 
-## 📝 使用场景
+### Training Procedure
 
-- 🔍 **搜索引擎优化 (SEO)** - 识别网站页面结构
-- 🕷️ **网页爬虫** - 判断链接类型，优化爬取策略
-- 📊 **网站分析** - 统计列表页和详情页比例
-- 🔗 **链接分类** - 大规模URL分类处理
+<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
 
-## 📁 相关链接
+#### Preprocessing [optional]
 
-- **GitHub仓库**: https://github.com/xiuxiu/url-classifier
-- **HuggingFace模型**: https://huggingface.co/windlx/url-classifier-model
-- **训练数据集**: https://huggingface.co/datasets/IowaCat/page_type_inference_dataset
+[More Information Needed]
 
-## 🙏 致谢
 
-- [Qwen](https://github.com/QwenLM/Qwen2) - 提供基础模型
-- [LoRA](https://github.com/microsoft/LoRA) - 高效微调方法
-- [HuggingFace](https://huggingface.co/) - 模型托管平台
+#### Training Hyperparameters
 
-## 📄 许可
+- **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
 
-[LICENSE](LICENSE) - MIT License
+#### Speeds, Sizes, Times [optional]
+
+<!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
+
+[More Information Needed]
+
+## Evaluation
+
+<!-- This section describes the evaluation protocols and provides the results. -->
+
+### Testing Data, Factors & Metrics
+
+#### Testing Data
+
+<!-- This should link to a Dataset Card if possible. -->
+
+[More Information Needed]
+
+#### Factors
+
+<!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
+
+[More Information Needed]
+
+#### Metrics
+
+<!-- These are the evaluation metrics being used, ideally with a description of why. -->
+
+[More Information Needed]
+
+### Results
+
+[More Information Needed]
+
+#### Summary
+
+
+
+## Model Examination [optional]
+
+<!-- Relevant interpretability work for the model goes here -->
+
+[More Information Needed]
+
+## Environmental Impact
+
+<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
+
+Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
+
+- **Hardware Type:** [More Information Needed]
+- **Hours used:** [More Information Needed]
+- **Cloud Provider:** [More Information Needed]
+- **Compute Region:** [More Information Needed]
+- **Carbon Emitted:** [More Information Needed]
+
+## Technical Specifications [optional]
+
+### Model Architecture and Objective
+
+[More Information Needed]
+
+### Compute Infrastructure
+
+[More Information Needed]
+
+#### Hardware
+
+[More Information Needed]
+
+#### Software
+
+[More Information Needed]
+
+## Citation [optional]
+
+<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
+
+**BibTeX:**
+
+[More Information Needed]
+
+**APA:**
+
+[More Information Needed]
+
+## Glossary [optional]
+
+<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
+
+[More Information Needed]
+
+## More Information [optional]
+
+[More Information Needed]
+
+## Model Card Authors [optional]
+
+[More Information Needed]
+
+## Model Card Contact
+
+[More Information Needed]
+### Framework versions
+
+- PEFT 0.17.1

adapter_config.json

@@ -25,13 +25,13 @@
   "rank_pattern": {},
   "revision": null,
   "target_modules": [
-    "v_proj",
     "k_proj",
-    "down_proj",
+    "up_proj",
     "q_proj",
     "o_proj",
+    "v_proj",
     "gate_proj",
-    "up_proj"
+    "down_proj"
   ],
   "target_parameters": null,
   "task_type": "CAUSAL_LM",

adapter_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a4acdea42620010d2624c8891ffed23957ea2f954f52655777dccd43b9630af8
+oid sha256:6471e71b043d9d5e44fc81e39173c9803dc98c6be2fa5a55d540cc1f1575db0c
 size 73911112

added_tokens.json

@@ -0,0 +1,24 @@
+{
+  "</tool_call>": 151658,
+  "<tool_call>": 151657,
+  "<|box_end|>": 151649,
+  "<|box_start|>": 151648,
+  "<|endoftext|>": 151643,
+  "<|file_sep|>": 151664,
+  "<|fim_middle|>": 151660,
+  "<|fim_pad|>": 151662,
+  "<|fim_prefix|>": 151659,
+  "<|fim_suffix|>": 151661,
+  "<|im_end|>": 151645,
+  "<|im_start|>": 151644,
+  "<|image_pad|>": 151655,
+  "<|object_ref_end|>": 151647,
+  "<|object_ref_start|>": 151646,
+  "<|quad_end|>": 151651,
+  "<|quad_start|>": 151650,
+  "<|repo_name|>": 151663,
+  "<|video_pad|>": 151656,
+  "<|vision_end|>": 151653,
+  "<|vision_pad|>": 151654,
+  "<|vision_start|>": 151652
+}

chat_template.jinja

@@ -0,0 +1,54 @@
+{%- if tools %}
+    {{- '<|im_start|>system\n' }}
+    {%- if messages[0]['role'] == 'system' %}
+        {{- messages[0]['content'] }}
+    {%- else %}
+        {{- 'You are a helpful assistant.' }}
+    {%- endif %}
+    {{- "\n\n# Tools\n\nYou may call one or more functions to assist with the user query.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>" }}
+    {%- for tool in tools %}
+        {{- "\n" }}
+        {{- tool | tojson }}
+    {%- endfor %}
+    {{- "\n</tools>\n\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\n<tool_call>\n{\"name\": <function-name>, \"arguments\": <args-json-object>}\n</tool_call><|im_end|>\n" }}
+{%- else %}
+    {%- if messages[0]['role'] == 'system' %}
+        {{- '<|im_start|>system\n' + messages[0]['content'] + '<|im_end|>\n' }}
+    {%- else %}
+        {{- '<|im_start|>system\nYou are a helpful assistant.<|im_end|>\n' }}
+    {%- endif %}
+{%- endif %}
+{%- for message in messages %}
+    {%- if (message.role == "user") or (message.role == "system" and not loop.first) or (message.role == "assistant" and not message.tool_calls) %}
+        {{- '<|im_start|>' + message.role + '\n' + message.content + '<|im_end|>' + '\n' }}
+    {%- elif message.role == "assistant" %}
+        {{- '<|im_start|>' + message.role }}
+        {%- if message.content %}
+            {{- '\n' + message.content }}
+        {%- endif %}
+        {%- for tool_call in message.tool_calls %}
+            {%- if tool_call.function is defined %}
+                {%- set tool_call = tool_call.function %}
+            {%- endif %}
+            {{- '\n<tool_call>\n{"name": "' }}
+            {{- tool_call.name }}
+            {{- '", "arguments": ' }}
+            {{- tool_call.arguments | tojson }}
+            {{- '}\n</tool_call>' }}
+        {%- endfor %}
+        {{- '<|im_end|>\n' }}
+    {%- elif message.role == "tool" %}
+        {%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != "tool") %}
+            {{- '<|im_start|>user' }}
+        {%- endif %}
+        {{- '\n<tool_response>\n' }}
+        {{- message.content }}
+        {{- '\n</tool_response>' }}
+        {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
+            {{- '<|im_end|>\n' }}
+        {%- endif %}
+    {%- endif %}
+{%- endfor %}
+{%- if add_generation_prompt %}
+    {{- '<|im_start|>assistant\n' }}
+{%- endif %}

optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:237a47a630887bed620ce8be916d7e98f376b70d5443cefd302892184065113d
+size 148047722

rng_state.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:726d0d4bb83272af9b45ee5c326944b754813bd1cbcb101424c851a7cad990d7
+size 14244

scaler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c852b1c266b81ad33622067d7dadb27814ab6638ea8e4f9cd67d3515606d043d
+size 988

scheduler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:75eb251ac935d5db6e2f7cdb5cf5207c336af119368fdfc4b26832ab78ba7599
+size 1064

trainer_state.json

@@ -0,0 +1,139 @@
+{
+  "best_global_step": null,
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 0.24,
+  "eval_steps": 500,
+  "global_step": 300,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.016,
+      "grad_norm": 1.1355500221252441,
+      "learning_rate": 0.000199392,
+      "loss": 2.2663,
+      "step": 20
+    },
+    {
+      "epoch": 0.032,
+      "grad_norm": 0.7390090227127075,
+      "learning_rate": 0.000198752,
+      "loss": 1.3075,
+      "step": 40
+    },
+    {
+      "epoch": 0.048,
+      "grad_norm": 0.8225588798522949,
+      "learning_rate": 0.00019811200000000002,
+      "loss": 1.2692,
+      "step": 60
+    },
+    {
+      "epoch": 0.064,
+      "grad_norm": 0.39007440209388733,
+      "learning_rate": 0.00019747200000000002,
+      "loss": 1.2632,
+      "step": 80
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.5560928583145142,
+      "learning_rate": 0.00019683200000000003,
+      "loss": 1.2401,
+      "step": 100
+    },
+    {
+      "epoch": 0.096,
+      "grad_norm": 0.6265106797218323,
+      "learning_rate": 0.000196192,
+      "loss": 1.2026,
+      "step": 120
+    },
+    {
+      "epoch": 0.112,
+      "grad_norm": 0.41324910521507263,
+      "learning_rate": 0.00019555200000000001,
+      "loss": 1.2117,
+      "step": 140
+    },
+    {
+      "epoch": 0.128,
+      "grad_norm": 0.6018273234367371,
+      "learning_rate": 0.000194912,
+      "loss": 1.243,
+      "step": 160
+    },
+    {
+      "epoch": 0.144,
+      "grad_norm": 0.7157287001609802,
+      "learning_rate": 0.000194272,
+      "loss": 1.2232,
+      "step": 180
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 0.318360835313797,
+      "learning_rate": 0.000193632,
+      "loss": 1.2553,
+      "step": 200
+    },
+    {
+      "epoch": 0.176,
+      "grad_norm": 0.37768101692199707,
+      "learning_rate": 0.000192992,
+      "loss": 1.2367,
+      "step": 220
+    },
+    {
+      "epoch": 0.192,
+      "grad_norm": 0.28085458278656006,
+      "learning_rate": 0.000192352,
+      "loss": 1.228,
+      "step": 240
+    },
+    {
+      "epoch": 0.208,
+      "grad_norm": 0.3715125024318695,
+      "learning_rate": 0.000191712,
+      "loss": 1.2367,
+      "step": 260
+    },
+    {
+      "epoch": 0.224,
+      "grad_norm": 0.5925875306129456,
+      "learning_rate": 0.000191072,
+      "loss": 1.2121,
+      "step": 280
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.5892595648765564,
+      "learning_rate": 0.000190432,
+      "loss": 1.2138,
+      "step": 300
+    }
+  ],
+  "logging_steps": 20,
+  "max_steps": 6250,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 5,
+  "save_steps": 100,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": false
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 9797016276172800.0,
+  "train_batch_size": 2,
+  "trial_name": null,
+  "trial_params": null
+}

training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d0ff569697c91cff7f682e3cfc25e55c28071865f1ed4367d3e61f66ff34c8dd
+size 5432