def create_features3(self,
data_list=None,
pad_id=None,
dim=None,
need_pad=True,
dtype=np.long):
if not data_list:
data_list: List = \
[ # Instance 1:
[[[0, 1, 0], [1, 0, 0]],
[[1, 0, 0], [0, 1, 0]]],
# Instance 2:
[[[1, 0, 0], [0, 1, 0]]],
# Instance 3:
[[[0, 1, 0]],
[[1, 0, 0], [1, 0, 0]],
[[0, 1, 0], [1, 0, 0]]]
]
pad_id: List = [0, 0, 1] if pad_id is None else pad_id
dim: int = 2 if dim is None else dim
features: List[Feature] = []
for data in data_list:
feature: Feature = Feature(data, {"pad_value": pad_id,
"dim": dim,
"dtype": dtype,
"need_pad": need_pad
})
features.append(feature)
return features
def create_features2(
self,
data_list=None,
pad_id=None,
dim=None,
need_pad=True,
dtype=np.long,
):
if not data_list:
data_list = [
[[6, 11, 2], [7, 8], [6, 7, 5, 4]],
[[4, 3], [7, 6]],
[[6, 3], [5], [7, 12], [7, 11]],
]
pad_id: int = 0 if pad_id is None else pad_id
dim: int = 2 if dim is None else dim
features: List[Feature] = []
for data in data_list:
feature: Feature = Feature(
data,
{
"pad_value": pad_id,
"dim": dim,
"dtype": dtype,
"need_pad": need_pad,
},
)
features.append(feature)
return features
def create_features1(self,
data_list=None,
pad_id=0,
dim=1,
need_pad=True,
dtype=np.long):
if not data_list:
data_list = [[7, 8, 9], [1, 2, 5, 6], [4]]
pad_id: int = 0 if pad_id is None else pad_id
dim: int = 1 if dim is None else dim
features: List[Feature] = []
for data in data_list:
feature: Feature = Feature(
data,
{
"pad_value": pad_id,
"dim": dim,
"dtype": dtype,
"need_pad": need_pad,
},
)
features.append(feature)
return features
def create_feature3(self,
data=None,
pad_id=None,
dim=None,
need_pad=True,
dtype=torch.long):
data: List = ([[[0, 1, 0], [1, 0, 0], [1, 0, 0]],
[[1, 0, 0], [0, 1, 0]]] if data is None else data)
pad_id: List = [0, 0, 1] if pad_id is None else pad_id
dim: int = 2 if dim is None else dim
feature: Feature = Feature(
data,
{
"pad_value": pad_id,
"dim": dim,
"dtype": dtype,
"need_pad": need_pad,
},
)
return feature
def create_feature1(self,
data=None,
pad_id=None,
dim=None,
need_pad=True,
dtype=None):
data: List = [7, 8, 9] if data is None else data
pad_id: int = 0 if pad_id is None else pad_id
dim: int = 1 if dim is None else dim
dtype = torch.long if dtype is None else dtype
feature: Feature = Feature(
data,
{
"pad_value": pad_id,
"dim": dim,
"dtype": dtype,
"need_pad": need_pad,
},
)
return feature
def create_feature2(self,
data=None,
pad_id=None,
dim=None,
need_pad=True,
dtype=None):
data: List = ([[6, 11, 2], [7, 8], [6, 7, 5, 4]]
if data is None else data)
pad_id: int = 0 if pad_id is None else pad_id
dim: int = 2 if dim is None else dim
dtype = torch.long if dtype is None else dtype
feature: Feature = Feature(
data,
{
"pad_value": pad_id,
"dim": dim,
"dtype": dtype,
"need_pad": need_pad,
},
)
return feature