科学数据
用法
Scidata当前支持以下训练和测试数据集:
MNIST
时尚MNIST
CIFAR10
在本地下载或获取数据集:
{train_images, train_labels} = Scidata . MNIST . download ()
{test_images, test_labels} = Scidata . MNIST . download_test ()
# Unpack train_images like...
{images_binary, tensor_type, shape} = train_images
您还可以将转换函数传递给download/1 :
transform_images = fn {binary, type, shape} ->
binary
|> Nx . from_binary (type)
|> Nx . r
评论0
最新资源