|
def | __init__ (self, squad_json, vocab_file, cache_file, batch_size, max_seq_length, num_inputs) |
|
def | free (self) |
|
def | get_batch_size (self) |
|
def | get_batch (self, names) |
|
def | read_calibration_cache (self) |
|
def | write_calibration_cache (self, cache) |
|
def | get_quantile (self) |
|
def | get_regression_cutoff (self) |
|
def | read_histogram_cache (self, length) |
|
def | write_histogram_cache (self, ptr, length) |
|
◆ __init__()
def helpers.calibrator.BertCalibrator.__init__ |
( |
|
self, |
|
|
|
squad_json, |
|
|
|
vocab_file, |
|
|
|
cache_file, |
|
|
|
batch_size, |
|
|
|
max_seq_length, |
|
|
|
num_inputs |
|
) |
| |
◆ free()
def helpers.calibrator.BertCalibrator.free |
( |
|
self | ) |
|
◆ get_batch_size()
def helpers.calibrator.BertCalibrator.get_batch_size |
( |
|
self | ) |
|
◆ get_batch()
def helpers.calibrator.BertCalibrator.get_batch |
( |
|
self, |
|
|
|
names |
|
) |
| |
◆ read_calibration_cache()
def helpers.calibrator.BertCalibrator.read_calibration_cache |
( |
|
self | ) |
|
◆ write_calibration_cache()
def helpers.calibrator.BertCalibrator.write_calibration_cache |
( |
|
self, |
|
|
|
cache |
|
) |
| |
◆ get_quantile()
def helpers.calibrator.BertCalibrator.get_quantile |
( |
|
self | ) |
|
◆ get_regression_cutoff()
def helpers.calibrator.BertCalibrator.get_regression_cutoff |
( |
|
self | ) |
|
◆ read_histogram_cache()
def helpers.calibrator.BertCalibrator.read_histogram_cache |
( |
|
self, |
|
|
|
length |
|
) |
| |
◆ write_histogram_cache()
def helpers.calibrator.BertCalibrator.write_histogram_cache |
( |
|
self, |
|
|
|
ptr, |
|
|
|
length |
|
) |
| |
◆ cache_file
helpers.calibrator.BertCalibrator.cache_file |
◆ data
helpers.calibrator.BertCalibrator.data |
◆ max_seq_length
helpers.calibrator.BertCalibrator.max_seq_length |
◆ batch_size
helpers.calibrator.BertCalibrator.batch_size |
◆ current_index
helpers.calibrator.BertCalibrator.current_index |
◆ num_inputs
helpers.calibrator.BertCalibrator.num_inputs |
◆ tokenizer
helpers.calibrator.BertCalibrator.tokenizer |
◆ doc_stride
helpers.calibrator.BertCalibrator.doc_stride |
◆ max_query_length
helpers.calibrator.BertCalibrator.max_query_length |
◆ device_inputs
helpers.calibrator.BertCalibrator.device_inputs |
The documentation for this class was generated from the following file:
- demo/BERT/helpers/calibrator.py