Convert sampathkethineedi/industry-classification-api to Transformers.js

config.json

@@ -0,0 +1,153 @@
+{
+  "_name_or_path": "sampathkethineedi/industry-classification-api",
+  "architectures": [
+    "BertForSequenceClassification"
+  ],
+  "attention_probs_dropout_prob": 0.1,
+  "classifier_dropout": null,
+  "gradient_checkpointing": false,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "Advertising",
+    "1": "Aerospace & Defense",
+    "2": "Apparel Retail",
+    "3": "Apparel, Accessories & Luxury Goods",
+    "4": "Application Software",
+    "5": "Asset Management & Custody Banks",
+    "6": "Auto Parts & Equipment",
+    "7": "Biotechnology",
+    "8": "Building Products",
+    "9": "Casinos & Gaming",
+    "10": "Commodity Chemicals",
+    "11": "Communications Equipment",
+    "12": "Construction & Engineering",
+    "13": "Construction Machinery & Heavy Trucks",
+    "14": "Consumer Finance",
+    "15": "Data Processing & Outsourced Services",
+    "16": "Diversified Metals & Mining",
+    "17": "Diversified Support Services",
+    "18": "Electric Utilities",
+    "19": "Electrical Components & Equipment",
+    "20": "Electronic Equipment & Instruments",
+    "21": "Environmental & Facilities Services",
+    "22": "Gold",
+    "23": "Health Care Equipment",
+    "24": "Health Care Facilities",
+    "25": "Health Care Services",
+    "26": "Health Care Supplies",
+    "27": "Health Care Technology",
+    "28": "Homebuilding",
+    "29": "Hotels, Resorts & Cruise Lines",
+    "30": "Human Resource & Employment Services",
+    "31": "IT Consulting & Other Services",
+    "32": "Industrial Machinery",
+    "33": "Integrated Telecommunication Services",
+    "34": "Interactive Media & Services",
+    "35": "Internet & Direct Marketing Retail",
+    "36": "Internet Services & Infrastructure",
+    "37": "Investment Banking & Brokerage",
+    "38": "Leisure Products",
+    "39": "Life Sciences Tools & Services",
+    "40": "Movies & Entertainment",
+    "41": "Oil & Gas Equipment & Services",
+    "42": "Oil & Gas Exploration & Production",
+    "43": "Oil & Gas Refining & Marketing",
+    "44": "Oil & Gas Storage & Transportation",
+    "45": "Packaged Foods & Meats",
+    "46": "Personal Products",
+    "47": "Pharmaceuticals",
+    "48": "Property & Casualty Insurance",
+    "49": "Real Estate Operating Companies",
+    "50": "Regional Banks",
+    "51": "Research & Consulting Services",
+    "52": "Restaurants",
+    "53": "Semiconductors",
+    "54": "Specialty Chemicals",
+    "55": "Specialty Stores",
+    "56": "Steel",
+    "57": "Systems Software",
+    "58": "Technology Distributors",
+    "59": "Technology Hardware, Storage & Peripherals",
+    "60": "Thrifts & Mortgage Finance",
+    "61": "Trading Companies & Distributors"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "Advertising": 0,
+    "Aerospace & Defense": 1,
+    "Apparel Retail": 2,
+    "Apparel, Accessories & Luxury Goods": 3,
+    "Application Software": 4,
+    "Asset Management & Custody Banks": 5,
+    "Auto Parts & Equipment": 6,
+    "Biotechnology": 7,
+    "Building Products": 8,
+    "Casinos & Gaming": 9,
+    "Commodity Chemicals": 10,
+    "Communications Equipment": 11,
+    "Construction & Engineering": 12,
+    "Construction Machinery & Heavy Trucks": 13,
+    "Consumer Finance": 14,
+    "Data Processing & Outsourced Services": 15,
+    "Diversified Metals & Mining": 16,
+    "Diversified Support Services": 17,
+    "Electric Utilities": 18,
+    "Electrical Components & Equipment": 19,
+    "Electronic Equipment & Instruments": 20,
+    "Environmental & Facilities Services": 21,
+    "Gold": 22,
+    "Health Care Equipment": 23,
+    "Health Care Facilities": 24,
+    "Health Care Services": 25,
+    "Health Care Supplies": 26,
+    "Health Care Technology": 27,
+    "Homebuilding": 28,
+    "Hotels, Resorts & Cruise Lines": 29,
+    "Human Resource & Employment Services": 30,
+    "IT Consulting & Other Services": 31,
+    "Industrial Machinery": 32,
+    "Integrated Telecommunication Services": 33,
+    "Interactive Media & Services": 34,
+    "Internet & Direct Marketing Retail": 35,
+    "Internet Services & Infrastructure": 36,
+    "Investment Banking & Brokerage": 37,
+    "Leisure Products": 38,
+    "Life Sciences Tools & Services": 39,
+    "Movies & Entertainment": 40,
+    "Oil & Gas Equipment & Services": 41,
+    "Oil & Gas Exploration & Production": 42,
+    "Oil & Gas Refining & Marketing": 43,
+    "Oil & Gas Storage & Transportation": 44,
+    "Packaged Foods & Meats": 45,
+    "Personal Products": 46,
+    "Pharmaceuticals": 47,
+    "Property & Casualty Insurance": 48,
+    "Real Estate Operating Companies": 49,
+    "Regional Banks": 50,
+    "Research & Consulting Services": 51,
+    "Restaurants": 52,
+    "Semiconductors": 53,
+    "Specialty Chemicals": 54,
+    "Specialty Stores": 55,
+    "Steel": 56,
+    "Systems Software": 57,
+    "Technology Distributors": 58,
+    "Technology Hardware, Storage & Peripherals": 59,
+    "Thrifts & Mortgage Finance": 60,
+    "Trading Companies & Distributors": 61
+  },
+  "layer_norm_eps": 1e-12,
+  "max_position_embeddings": 512,
+  "model_type": "bert",
+  "num_attention_heads": 12,
+  "num_hidden_layers": 12,
+  "pad_token_id": 0,
+  "position_embedding_type": "absolute",
+  "transformers_version": "4.30.2",
+  "type_vocab_size": 2,
+  "use_cache": true,
+  "vocab_size": 30522
+}

onnx/model.onnx

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

onnx/model_quantized.onnx

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

quantize_config.json

@@ -0,0 +1,32 @@
+{
+    "per_channel": true,
+    "reduce_range": true,
+    "per_model_config": {
+        "model": {
+            "op_types": [
+                "Mul",
+                "Slice",
+                "Div",
+                "Sub",
+                "Erf",
+                "Concat",
+                "Pow",
+                "Cast",
+                "Transpose",
+                "ReduceMean",
+                "MatMul",
+                "Reshape",
+                "Constant",
+                "Tanh",
+                "Softmax",
+                "Shape",
+                "Gemm",
+                "Sqrt",
+                "Add",
+                "Gather",
+                "Unsqueeze"
+            ],
+            "weight_type": "QInt8"
+        }
+    }
+}

special_tokens_map.json

@@ -0,0 +1,7 @@
+{
+  "cls_token": "[CLS]",
+  "mask_token": "[MASK]",
+  "pad_token": "[PAD]",
+  "sep_token": "[SEP]",
+  "unk_token": "[UNK]"
+}

tokenizer_config.json

@@ -0,0 +1,15 @@
+{
+  "clean_up_tokenization_spaces": true,
+  "cls_token": "[CLS]",
+  "do_basic_tokenize": true,
+  "do_lower_case": true,
+  "mask_token": "[MASK]",
+  "model_max_length": 512,
+  "never_split": null,
+  "pad_token": "[PAD]",
+  "sep_token": "[SEP]",
+  "strip_accents": null,
+  "tokenize_chinese_chars": true,
+  "tokenizer_class": "BertTokenizer",
+  "unk_token": "[UNK]"
+}