d35b8df03752363a70ddf4d30dda099d3e016958,examples/code_reuse_example/code_reuse_example.py,,,#,30

Before Change

test_y = encoder.transform(test_y)

compose_list = Compose([])
train_data = DataLoader(MultiTransformDataset(torch.Tensor(train_x), torch.Tensor(train_y), compose_list), batch_size=batch_size, shuffle=False)
test_data = DataLoader(MultiTransformDataset(torch.Tensor(test_x), torch.Tensor(test_y), compose_list), batch_size=batch_size, shuffle=False)

model_trainer = ModelTrainer(model,
                             loss_function=classification_loss,
                             metric=Accuracy,
                             train_data=train_data,

After Change

encoder = OneHotEncoder()
encoder.fit(train_y)
train_y = torch.Tensor(encoder.transform(train_y))
test_y = torch.Tensor(encoder.transform(test_y))

compose_list = Compose([])
train_data = DataLoader(MultiTransformDataset(train_x, train_y, compose_list),
                        batch_size=batch_size,
                        shuffle=False)
test_data = DataLoader(MultiTransformDataset(test_x, test_y, compose_list),
                       batch_size=batch_size,
                       shuffle=False)

model_trainer = ModelTrainer(model,
                             loss_function=classification_loss,
                             metric=Accuracy,
                             train_data=train_data,

Instances