feat(julia): add tests for new changes

This commit is contained in:
Amaan Qureshi 2023-03-09 19:58:04 -05:00
parent 1012dd957e
commit ffa6efa7f5

View file

@ -0,0 +1,26 @@
function load_data(::Symbol; ::Int) :: Tuple
# <- keyword.function
# ^ function
# ^ punctuation.bracket
# ^^ punctuation.delimiter
# ^ type.builtin
# ^ punctuation.delimiter
# ^^ punctuation.delimiter
# ^^^ type.builtin
# ^ punctuation.bracket
# ^^ punctuation.delimiter
# ^ type.builtin
dataset = CIFAR10(; Tx = Float32, split = split)
# ^^^^^^^ variable
# ^ operator
# ^ function.call
# ^ operator
# ^ type.builtin
X = reshape(dataset.features[:, :, :, begin:n_obs], :, n_obs) # flattening the image pixels
# ^^^^^ variable.builtin
y = categorical2onehot(dataset.targets[begin:n_obs], N_LABELS)
# ^^^^^ variable.builtin
return X, y
# ^^^^^^ keyword.return
end
# <- keyword.function