task preprocessing is working

mlr-org · Oct 18, 2023 · d7df862 · d7df862
1 parent e29f6b6
commit d7df862
Show file tree

Hide file tree

Showing 21 changed files with 815 additions and 446 deletions.
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -113,7 +113,6 @@ Collate:
     'PipeOpTorchOptimizer.R'
     'PipeOpTorchReshape.R'
     'PipeOpTorchSoftmax.R'
-    'PipeOpTorchTrafo.R'
     'ResamplingRowRoles.R'
     'TaskClassif_mnist.R'
     'TaskClassif_tiny_imagenet.R'
@@ -128,8 +127,7 @@ Collate:
     'merge_graphs.R'
     'nn_graph.R'
     'paramset_torchlearner.R'
-    'register_lazy.R'
-    'preproc_and_trafo.R'
+    'preprocess.R'
     'rd_info.R'
     'reset_last_layer.R'
     'task_dataset.R'

diff --git a/NAMESPACE b/NAMESPACE
@@ -127,7 +127,6 @@ export(PipeOpTorchSqueeze)
 export(PipeOpTorchTanh)
 export(PipeOpTorchTanhShrink)
 export(PipeOpTorchThreshold)
-export(PipeOpTorchTrafo)
 export(PipeOpTorchUnsqueeze)
 export(ResamplingRowRoles)
 export(TorchCallback)
@@ -160,6 +159,7 @@ export(nn_merge_sum)
 export(nn_reshape)
 export(nn_squeeze)
 export(nn_unsqueeze)
+export(pipeop_preproc_torch)
 export(reset_last_layer)
 export(t_clbk)
 export(t_clbks)
@@ -170,8 +170,7 @@ export(t_opts)
 export(task_dataset)
 export(torch_callback)
 export(transform_lazy_tensor)
-exportPattern("^PipeOpTaskPreprocTorch")
-exportPattern("^PipeOpTorchTrafo")
+exportPattern("^PipeOpPreproc")
 import(checkmate)
 import(data.table)
 import(mlr3)