draugr/samples/torch_samples/cross_valid_example.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

__author__ = "Christian Heider Nielsen"
__doc__ = r"""

           Created on 31-10-2020
           """
import torch
from torch.utils.data import TensorDataset

from draugr.torch_utilities import cross_validation_generator, to_tensor


def asdasidoj() -> None:
    """
    :rtype: None
    """
    X = to_tensor([torch.diag(torch.arange(i, i + 2)) for i in range(200)])
    x_train = TensorDataset(X[:100])
    x_val = TensorDataset(X[100:])

    for train, val in cross_validation_generator(x_train, x_val):
        print(len(train), len(val))
        print(train[0], val[0])


asdasidoj()