Files
qwen3-4b-eventspec-martech-…/README.md

168 lines
7.5 KiB
Markdown
Raw Normal View History

---
license: apache-2.0
base_model: Qwen/Qwen3-4B-Instruct-2507
tags:
- martech
- analytics
- event-taxonomy
- json
- structured-output
- sft
- lora
library_name: transformers
pipeline_tag: text-generation
---
# Qwen3-4B EventSpec — MarTech Event Taxonomy Generator (merged)
Fine-tuned **Qwen/Qwen3-4B-Instruct-2507** that converts free-form marketing tracking
requests into clean, implementation-ready **analytics event specifications** as strict JSON.
This repository contains the **fully merged weights** (LoRA adapter merged into the base
model), so it loads like any standard model — no PEFT/adapter step required.
- **Base model:** [Qwen/Qwen3-4B-Instruct-2507](https://huggingface.co/Qwen/Qwen3-4B-Instruct-2507)
- **LoRA adapter (pre-merge):** [joshelu/qwen3-4b-Instruct-eventspec-martech-sft](https://huggingface.co/joshelu/qwen3-4b-Instruct-eventspec-martech-sft)
- **Method:** Supervised Fine-Tuning (SFT) + LoRA, then merged
- **Task:** marketing tracking request → strict-JSON EventSpec
## Intended use
Give the model a plain-English marketing/analytics tracking request; it returns a single
JSON object specifying the events to implement (event names, parameters, triggers, consent
and deduplication requirements, QA criteria, risk flags, etc.). Useful for MarTech / analytics
engineering, GA4 / GTM instrumentation planning, and taxonomy standardization.
## ⚠️ Prompt format (required for correct output)
The model was trained with a specific chat format. **You must reproduce it** or output
quality degrades sharply.
**System prompt (use verbatim):**
```
You are EventSpec, an expert MarTech analytics engineer. You convert free-form marketing tracking requests into clean, implementation-ready analytics event specifications.
Given a marketing tracking request, respond with a SINGLE valid JSON object and nothing else: no prose, no markdown, no code fences. The JSON must be strictly parseable.
The specification captures: a concise `request_summary`; the `business_goal`; the `tracking_scope` (platforms, page_or_screen, user_action, conversion_type); and a list of `recommended_events`. Each recommended event defines `event_name` (snake_case), `event_description`, `platform`, `event_type`, `required_parameters`, `optional_parameters`, `trigger_condition`, `consent_requirements`, and `deduplication_requirements`.
Use consistent snake_case event and parameter names, follow analytics best practices (GA4/GTM conventions where relevant), and respect privacy/consent requirements. Output only the JSON object.
```
**User message format:**
```
Convert this marketing tracking request into a clean analytics event specification.
Request:
<your tracking request here>
```
## Usage (transformers)
```python
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
repo = "joshelu/qwen3-4b-eventspec-martech-merged"
tok = AutoTokenizer.from_pretrained(repo)
model = AutoModelForCausalLM.from_pretrained(repo, dtype=torch.bfloat16, device_map="auto")
SYSTEM_PROMPT = (
"You are EventSpec, an expert MarTech analytics engineer. You convert free-form "
"marketing tracking requests into clean, implementation-ready analytics event "
"specifications.\n\n"
"Given a marketing tracking request, respond with a SINGLE valid JSON object and "
"nothing else: no prose, no markdown, no code fences. The JSON must be strictly "
"parseable.\n\n"
"The specification captures: a concise `request_summary`; the `business_goal`; the "
"`tracking_scope` (platforms, page_or_screen, user_action, conversion_type); and a "
"list of `recommended_events`. Each recommended event defines `event_name` "
"(snake_case), `event_description`, `platform`, `event_type`, `required_parameters`, "
"`optional_parameters`, `trigger_condition`, `consent_requirements`, and "
"`deduplication_requirements`.\n\n"
"Use consistent snake_case event and parameter names, follow analytics best "
"practices (GA4/GTM conventions where relevant), and respect privacy/consent "
"requirements. Output only the JSON object."
)
USER_PREFIX = "Convert this marketing tracking request into a clean analytics event specification."
request = ("A pharmacy app wants to track refill reminders, refill started, refill submitted, "
"refill ready, and pickup completed, but no medication names or prescription numbers "
"should be sent.")
messages = [
{"role": "system", "content": SYSTEM_PROMPT},
{"role": "user", "content": f"{USER_PREFIX}\n\nRequest:\n{request}"},
]
prompt = tok.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
inputs = tok(prompt, return_tensors="pt").to(model.device)
out = model.generate(**inputs, max_new_tokens=2048, do_sample=False,
pad_token_id=tok.eos_token_id)
print(tok.decode(out[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True))
```
## Usage (Inference Endpoints / chat_completion)
```python
from huggingface_hub import InferenceClient
client = InferenceClient("https://YOUR-ENDPOINT.endpoints.huggingface.cloud", token="hf_...")
resp = client.chat_completion(
messages=[
{"role": "system", "content": SYSTEM_PROMPT},
{"role": "user", "content": f"{USER_PREFIX}\n\nRequest:\n{request}"},
],
max_tokens=2048,
temperature=0,
)
print(resp.choices[0].message.content)
```
## Recommended generation settings
- `temperature = 0` (greedy) — best for stable, strictly parseable JSON.
- `max_new_tokens >= 2048` — specs are long; a lower limit will truncate the JSON mid-string.
- Parse the output with `json.loads`; retry with a higher token limit if parsing fails.
## Output schema
The model produces a single JSON object. Core keys:
- `request_summary` — one-line summary of the request.
- `business_goal` — the measurement objective.
- `tracking_scope``{ platforms, page_or_screen, user_action, conversion_type }`.
- `recommended_events` — list of events, each with `event_name` (snake_case),
`event_description`, `platform`, `event_type`, `required_parameters`,
`optional_parameters`, `trigger_condition`, `consent_requirements`,
`deduplication_requirements`.
Richer examples in the training data also include `implementation_notes`, `qa_criteria`,
`open_questions`, and `risk_flags`, which the model produces as appropriate.
## Training
- **Dataset:** `joshelu/martech-event-taxonomy-mapper-training-data` (private).
- The 100-row `train` split, **minus** every `id` appearing in the `validation` or `test`
split → **80 training rows** (held-out eval stays honest).
- Eval during training used the `validation` split (10 rows).
- Assistant targets are compact, strict JSON (`json.dumps(output, separators=(",", ":"))`),
validated as parseable before training.
- **Method:** SFT + LoRA, then merged.
- LoRA: `r=16`, `alpha=32`, `dropout=0.05`,
target modules `q_proj, k_proj, v_proj, o_proj, gate_proj, up_proj, down_proj`.
- 3 epochs, effective batch size 4 (per-device 1 × grad-accum 4), lr `2e-4`,
warmup ratio `0.05`, cosine schedule, max length 2048, bf16, gradient checkpointing.
- Hardware: a10g-small (HF Jobs).
- **Metrics (final):** eval loss ≈ `0.320`, eval mean token accuracy ≈ `0.924`.
## Limitations
- Trained on a small (80-row) dataset — coverage is limited to the taxonomy and styles seen
in training; unusual domains may produce weaker specs.
- Output is a strong **draft**, not a substitute for review by an analytics engineer,
especially for consent/privacy and PII handling.
- With very low `max_new_tokens`, long specs will be truncated and fail JSON parsing — keep
the limit high and validate the parse.