Text Classification
Transformers
Safetensors
English
roberta
record-types
data-licensing
company-data
data-export
text-embeddings-inference
aitrainingdatabroker commited on
Commit
bb289ee
·
verified ·
1 Parent(s): 41eb0bf

Add README.md

Browse files
Files changed (1) hide show
  1. README.md +102 -0
README.md ADDED
@@ -0,0 +1,102 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ library_name: transformers
6
+ pipeline_tag: text-classification
7
+ base_model: distilroberta-base
8
+ datasets:
9
+ - aitrainingdatabroker/company-record-types-and-limits
10
+ - aitrainingdatabroker/company-data-licensing-faq-by-topic
11
+ tags:
12
+ - record-types
13
+ - data-licensing
14
+ - text-classification
15
+ - company-data
16
+ - data-export
17
+ ---
18
+
19
+ # Record Type Page Classifier
20
+
21
+ Reads a passage about company records and says which of 33 record types it is most likely about: support tickets, team chat, CRM history, CAD files, tool exports such as Slack or Jira, and more.
22
+
23
+ All training text is our own, from pages published at aitrainingdatabroker.com. Nothing comes from a customer.
24
+
25
+ Related tool: [company-record-types-and-limits-finder](https://huggingface.co/spaces/aitrainingdatabroker/company-record-types-and-limits-finder) (a Space that runs in your browser).
26
+
27
+ ## Why it exists
28
+
29
+ Before data is prepared, a team has to say what kind of record each file is. This model suggests the record type for a passage so a person can confirm it.
30
+
31
+ ## Use it
32
+
33
+ ```python
34
+ from transformers import pipeline
35
+
36
+ classify = pipeline("text-classification", model="aitrainingdatabroker/record-type-page-classifier", top_k=3)
37
+ print(classify("A Jira export with issue keys, comments, status changes and linked pull requests."))
38
+ ```
39
+
40
+ ## How it was trained
41
+
42
+ Fine-tuned from `distilroberta-base` for 25 epochs on one RTX 3070. Training data: 419 passages from [company-record-types-and-limits](https://huggingface.co/datasets/aitrainingdatabroker/company-record-types-and-limits), [company-data-licensing-faq-by-topic](https://huggingface.co/datasets/aitrainingdatabroker/company-data-licensing-faq-by-topic), which are our own published page text. It was trained on item descriptions, introductions and section paragraphs from our 33 record-type pages and tested on the questions and answers of those pages, which it never saw in training. The scripts that built the data and trained the model are in `source/`; the full scores are in `scores.json`.
43
+
44
+ ## Test scores
45
+
46
+ On 107 held-out rows: accuracy 0.533, macro F1 0.497.
47
+
48
+ | Label | F1 | Count |
49
+ |---|---|---|
50
+ | `accounting-and-erp-records` | 0.0 | 3 |
51
+ | `cad-and-engineering-files` | 0.8 | 3 |
52
+ | `call-recordings` | 0.571 | 3 |
53
+ | `change-orders-and-rfis` | 0.8 | 3 |
54
+ | `claims-and-underwriting-files` | 0.5 | 3 |
55
+ | `client-deliverables-and-drafts` | 0.4 | 3 |
56
+ | `closed-deal-files` | 0.5 | 3 |
57
+ | `code-and-pull-request-history` | 0.0 | 3 |
58
+ | `confluence-data-export` | 0.889 | 4 |
59
+ | `consented-recordings` | 0.667 | 3 |
60
+ | `crm-history` | 0.5 | 3 |
61
+ | `egocentric-video` | 0.857 | 4 |
62
+ | `github-repository-export` | 0.444 | 3 |
63
+ | `gmail-data-export` | 0.0 | 4 |
64
+ | `google-drive-export` | 0.462 | 3 |
65
+ | `hubspot-data-export` | 0.286 | 3 |
66
+ | `inspection-reports-and-photos` | 0.857 | 3 |
67
+ | `jira-data-export` | 0.333 | 3 |
68
+ | `lab-notebooks-and-experiment-records` | 0.667 | 3 |
69
+ | `labeled-tables-and-datasets` | 1.0 | 4 |
70
+ | `microsoft-365-email-export` | 0.364 | 3 |
71
+ | `native-word-excel-and-powerpoint-files` | 0.4 | 4 |
72
+ | `project-and-task-history` | 0.5 | 3 |
73
+ | `quickbooks-data-export` | 0.4 | 3 |
74
+ | `quotes-and-estimates` | 0.667 | 3 |
75
+ | `salesforce-data-export` | 0.0 | 4 |
76
+ | `slack-data-export` | 0.727 | 4 |
77
+ | `sops-and-knowledge-bases` | 0.667 | 3 |
78
+ | `support-tickets` | 0.571 | 3 |
79
+ | `team-chat` | 0.0 | 3 |
80
+ | `theses-and-dissertations` | 0.667 | 4 |
81
+ | `work-orders-and-dispatch-notes` | 0.5 | 3 |
82
+ | `zendesk-data-export` | 0.4 | 3 |
83
+
84
+ ## Limits
85
+
86
+ - 33 classes with about 13 training passages each, and several are close neighbours (the export guides especially), so treat the top three as candidates.
87
+ - Trained on one broker's page text.
88
+ - This model is not legal advice and not a certification. A person should review its output before anything is shared or decided, and your counsel decides what a rule requires.
89
+
90
+ ## Read more
91
+
92
+ - [Record types and export guides](https://aitrainingdatabroker.com/data/?utm_source=huggingface&utm_medium=model&utm_campaign=record-type-page-classifier)
93
+ - [Can you license CRM history to AI?](https://aitrainingdatabroker.com/data/crm-history/?utm_source=huggingface&utm_medium=model&utm_campaign=record-type-page-classifier)
94
+ - [Jira data export for AI licensing](https://aitrainingdatabroker.com/data/jira-data-export/?utm_source=huggingface&utm_medium=model&utm_campaign=record-type-page-classifier)
95
+
96
+ ## License and credit
97
+
98
+ Apache 2.0. Please credit **AI Training Data Broker** and link to https://aitrainingdatabroker.com/ when you share or build on it.
99
+
100
+ ## About
101
+
102
+ AI Training Data Broker is a broker and technical advisor for companies that sell or license their data to AI labs. It guides the technical side, including export, format, de-identification and compliance, and publishes free guides, checklists and tools.