Skip to content

Repunt new model + quantization

Latest
Compare
Choose a tag to compare
@sviperm sviperm released this 08 Jul 15:59
· 1 commit to master since this release

Zip archive contains params.json with training params (we need this for configure our model's network), weights (*.pt) and logs.
Just extract this archive in models/ directory.
Expected directory structure after extraction:

models
└── repunct-model-new
    ├── logs
    │   └── repunct-model-new_logs.txt
    ├── params.json
    └── weights
        ├── weights_ep6_9912.pt
        └── _quantized_weights_ep5_9912.pt

Model evaluation metrics

Best validation Acc: 9912204619275377

Confusion Matrix:
	[22035741   102978     4773]
	[  107477  2112209     2878]
	[    7320     1558  1478818]
O:
	Precision: 0.9948
	Recall: 0.9951
	F1 score: 0.995
COMMA:
	Precision: 0.9528
	Recall: 0.9503
	F1 score: 0.9516
PERIOD:
	Precision: 0.9949
	Recall: 0.994
	F1 score: 0.9944
COMMA + PERIOD:
	Precision: 0.9697
	Recall: 0.9679
	F1 score: 0.9688