diff --git a/src/LoadSave/data_loaders.jl b/src/LoadSave/data_loaders.jl index a4620512..15841a5e 100644 --- a/src/LoadSave/data_loaders.jl +++ b/src/LoadSave/data_loaders.jl @@ -1,4 +1,7 @@ -export mnist, sampled_mnist, twenty_datasets, twenty_dataset_names, dataset +export mnist, + sampled_mnist, + twenty_datasets, + twenty_dataset_names using MLDatasets using CSV diff --git a/src/Utils/data.jl b/src/Utils/data.jl index 30f74dc2..560e2e7a 100644 --- a/src/Utils/data.jl +++ b/src/Utils/data.jl @@ -6,7 +6,7 @@ import CUDA: CuVector, CuMatrix export num_examples, num_features, example, feature_values, iscomplete, isfpdata, isbinarydata, - num_chunks, chunk, eltype, + num_chunks, chunks, eltype, shuffle_examples, batch, threshold, soften, to_gpu, to_cpu, isgpu, same_device, ll_per_example, bits_per_pixel diff --git a/src/bit_circuit.jl b/src/bit_circuit.jl index 4769f1c1..5220cf0c 100644 --- a/src/bit_circuit.jl +++ b/src/bit_circuit.jl @@ -1,7 +1,8 @@ using CUDA export NodeIds, ⋁NodeIds, ⋀NodeIds, BitCircuit, - num_nodes, num_decisions, num_elements, num_leafs, num_features, nodes, elements, + num_nodes, num_elements, num_leafs, num_features, + nodes, elements, has_single_child, sibling ##################### diff --git a/src/structured/abstract_vtrees.jl b/src/structured/abstract_vtrees.jl index 2ec69137..5549e35d 100644 --- a/src/structured/abstract_vtrees.jl +++ b/src/structured/abstract_vtrees.jl @@ -1,7 +1,7 @@ export Vtree, vtree, variable, find_leaf, - varsubset, varsubset_left, varsubset_right, lca_vtree, depth, - respects_vtree, - top_down_vtree, bottom_up_vtree + varsubset, varsubset_left, varsubset_right, + depth, + respects_vtree import Base: parent, in # extend import .Utils: find_leaf, find_inode, lca, depth # extend