Module lib.data

Expand source code 
from typing import NamedTuple

from jax import Array

class TrainData(NamedTuple):
    seq: Array
    seq_mask: Array
    labels: Array
    labels_mask: Array

class TestData(NamedTuple):
    seq: Array
    seq_mask: Array
    labels: list[str]

Classes

class TrainData (seq: jax.Array, seq_mask: jax.Array, labels: jax.Array, labels_mask: jax.Array)

TrainData(seq, seq_mask, labels, labels_mask)

Expand source code 
class TrainData(NamedTuple):
    seq: Array
    seq_mask: Array
    labels: Array
    labels_mask: Array

Ancestors

  • builtins.tuple

Instance variables

var seq : jax.Array

Alias for field number 0

var seq_mask : jax.Array

Alias for field number 1

var labels : jax.Array

Alias for field number 2

var labels_mask : jax.Array

Alias for field number 3

class TestData (seq: jax.Array, seq_mask: jax.Array, labels: list[str])

TestData(seq, seq_mask, labels)

Expand source code 
class TestData(NamedTuple):
    seq: Array
    seq_mask: Array
    labels: list[str]

Ancestors

  • builtins.tuple

Instance variables

var seq : jax.Array

Alias for field number 0

var seq_mask : jax.Array

Alias for field number 1

var labels : list[str]

Alias for field number 2