finipher-tax-document-classifier / tax_document_head_config.json
finipher's picture
v1.0.0
2f41bc6 verified
Raw History Blame Contribute Delete
5.22 kB
{
"artifact_version": "tax-document-head-20261027-v1.1-category-encoder",
"architecture": "frozen transaction MiniLM encoder + 95-way tax document head",
"base_model": "sentence-transformers/all-MiniLM-L6-v2",
"base_model_safetensors_sha256": "96d95c2520f9b8cee0a3e85e1440f541420c4c043e18a0c4579943e08e91a5a3",
"dataset": "finipher/finipher-us-tax-document-classifier",
"dataset_revision": "92c67c932c479e5f5944e5b5687aa327569e5de7",
"dataset_license": "CC-BY-4.0",
"num_labels": 95,
"hidden_size": 384,
"max_length": 128,
"pooling": "attention_mask_mean_l2_normalized",
"logit_scale_baked_into_weight": 10.0,
"selected_weight_decay": 1e-05,
"encoder_frozen": true,
"transaction_head_frozen": true,
"review_threshold": 0.76,
"label2id": {
"1040": 0,
"1040-ES": 1,
"1040-NR": 2,
"1040-SR": 3,
"1040-X": 4,
"1041": 5,
"1065": 6,
"1095-A": 7,
"1095-B": 8,
"1095-C": 9,
"1098": 10,
"1098-C": 11,
"1098-E": 12,
"1098-T": 13,
"1099-A": 14,
"1099-B": 15,
"1099-C": 16,
"1099-DA": 17,
"1099-DIV": 18,
"1099-G": 19,
"1099-INT": 20,
"1099-K": 21,
"1099-MISC": 22,
"1099-NEC": 23,
"1099-OID": 24,
"1099-Q": 25,
"1099-R": 26,
"1099-S": 27,
"1099-SA": 28,
"1116": 29,
"1120-S": 30,
"2441": 31,
"2555": 32,
"2848": 33,
"3520": 34,
"4506-T": 35,
"4868": 36,
"5498": 37,
"5498-ESA": 38,
"5498-SA": 39,
"5695": 40,
"709": 41,
"709-NA": 42,
"8283": 43,
"8332": 44,
"8889": 45,
"8936": 46,
"8938": 47,
"8949": 48,
"8962": 49,
"ANNUAL_GIVING_STATEMENT": 50,
"BANK_STATEMENT": 51,
"BUSINESS_SALES_LEDGER": 52,
"CHARITABLE_PLEDGE_NOTICE": 53,
"CHARITY_EVENT_RECEIPT": 54,
"CREDIT_CARD_STATEMENT": 55,
"DONATION_APPEAL": 56,
"ESTIMATED_TAX_PAYMENT_CONFIRMATION": 57,
"FOREIGN_GIFT_RECORD": 58,
"GIFT_TRANSFER_AGREEMENT": 59,
"K-1-1041": 60,
"K-1-1065": 61,
"K-1-1120-S": 62,
"MEDICAL_RECEIPT": 63,
"NONCASH_DONATION_ACKNOWLEDGMENT": 64,
"PERSONAL_GIFT_LETTER": 65,
"PURCHASE_RECEIPT": 66,
"RENTAL_INCOME_RECORD": 67,
"SCHEDULE-A": 68,
"SCHEDULE-B": 69,
"SCHEDULE-C": 70,
"SCHEDULE-D": 71,
"SCHEDULE-E": 72,
"SCHEDULE-F": 73,
"SCHEDULE-SE": 74,
"TAX_INSTRUCTIONS": 75,
"UNKNOWN_DOCUMENT": 76,
"VEHICLE_DONATION_ACKNOWLEDGMENT": 77,
"W-2": 78,
"W-2C": 79,
"W-2G": 80,
"W-4": 81,
"W-9": 82,
"MORTGAGE_STATEMENT": 83,
"LOAN_STATEMENT": 84,
"BROKERAGE_STATEMENT": 85,
"RETIREMENT_ACCOUNT_STATEMENT": 86,
"HSA_ACCOUNT_STATEMENT": 87,
"PAYSTUB": 88,
"INSURANCE_DOCUMENT": 89,
"CRYPTO_TAX_REPORT": 90,
"PROPERTY_TAX_BILL": 91,
"PROFIT_AND_LOSS_STATEMENT": 92,
"CONSOLIDATED_1099": 93,
"OTHER_TAX_FORM": 94
},
"id2label": {
"0": "1040",
"1": "1040-ES",
"2": "1040-NR",
"3": "1040-SR",
"4": "1040-X",
"5": "1041",
"6": "1065",
"7": "1095-A",
"8": "1095-B",
"9": "1095-C",
"10": "1098",
"11": "1098-C",
"12": "1098-E",
"13": "1098-T",
"14": "1099-A",
"15": "1099-B",
"16": "1099-C",
"17": "1099-DA",
"18": "1099-DIV",
"19": "1099-G",
"20": "1099-INT",
"21": "1099-K",
"22": "1099-MISC",
"23": "1099-NEC",
"24": "1099-OID",
"25": "1099-Q",
"26": "1099-R",
"27": "1099-S",
"28": "1099-SA",
"29": "1116",
"30": "1120-S",
"31": "2441",
"32": "2555",
"33": "2848",
"34": "3520",
"35": "4506-T",
"36": "4868",
"37": "5498",
"38": "5498-ESA",
"39": "5498-SA",
"40": "5695",
"41": "709",
"42": "709-NA",
"43": "8283",
"44": "8332",
"45": "8889",
"46": "8936",
"47": "8938",
"48": "8949",
"49": "8962",
"50": "ANNUAL_GIVING_STATEMENT",
"51": "BANK_STATEMENT",
"52": "BUSINESS_SALES_LEDGER",
"53": "CHARITABLE_PLEDGE_NOTICE",
"54": "CHARITY_EVENT_RECEIPT",
"55": "CREDIT_CARD_STATEMENT",
"56": "DONATION_APPEAL",
"57": "ESTIMATED_TAX_PAYMENT_CONFIRMATION",
"58": "FOREIGN_GIFT_RECORD",
"59": "GIFT_TRANSFER_AGREEMENT",
"60": "K-1-1041",
"61": "K-1-1065",
"62": "K-1-1120-S",
"63": "MEDICAL_RECEIPT",
"64": "NONCASH_DONATION_ACKNOWLEDGMENT",
"65": "PERSONAL_GIFT_LETTER",
"66": "PURCHASE_RECEIPT",
"67": "RENTAL_INCOME_RECORD",
"68": "SCHEDULE-A",
"69": "SCHEDULE-B",
"70": "SCHEDULE-C",
"71": "SCHEDULE-D",
"72": "SCHEDULE-E",
"73": "SCHEDULE-F",
"74": "SCHEDULE-SE",
"75": "TAX_INSTRUCTIONS",
"76": "UNKNOWN_DOCUMENT",
"77": "VEHICLE_DONATION_ACKNOWLEDGMENT",
"78": "W-2",
"79": "W-2C",
"80": "W-2G",
"81": "W-4",
"82": "W-9",
"83": "MORTGAGE_STATEMENT",
"84": "LOAN_STATEMENT",
"85": "BROKERAGE_STATEMENT",
"86": "RETIREMENT_ACCOUNT_STATEMENT",
"87": "HSA_ACCOUNT_STATEMENT",
"88": "PAYSTUB",
"89": "INSURANCE_DOCUMENT",
"90": "CRYPTO_TAX_REPORT",
"91": "PROPERTY_TAX_BILL",
"92": "PROFIT_AND_LOSS_STATEMENT",
"93": "CONSOLIDATED_1099",
"94": "OTHER_TAX_FORM"
}
}