MCPcopy Create free account
hub / github.com/NVIDIA/FasterTransformer / InputFeatures

Class InputFeatures

examples/tensorflow/xlnet/convertInput.py:393–406  ·  view source on GitHub ↗

A single set of features of data.

Source from the content-addressed store, hash-verified

391
392# Convert functions
393class InputFeatures(object):
394 """A single set of features of data."""
395
396 def __init__(self,
397 input_ids,
398 input_mask,
399 segment_ids,
400 label_id,
401 is_real_example=True):
402 self.input_ids = input_ids
403 self.input_mask = input_mask
404 self.segment_ids = segment_ids
405 self.label_id = label_id
406 self.is_real_example = is_real_example
407
408
409def _truncate_seq_pair(tokens_a, tokens_b, max_length):

Callers 1

convert_single_exampleFunction · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected