tdpeuter/2025ML-project-neural_compression
tdpeuter/2025ML-project-neural_compression
Archived
1
Fork 0
Code Issues Pull requests Releases 1 Activity Actions

feat: i think i set up the encoder

Browse source
This commit is contained in:
Robin Meersman 2025-11-08 17:40:49 +01:00
parent 947aba31ee
commit 63d1b6f5ae
2 changed files with 51 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

2
CNN-model/data_utils.py
View file

@ -3,7 +3,7 @@ from torch.utils.data import TensorDataset
def make_context_pairs(data: bytes, context_length: int) -> TensorDataset:
data = torch.tensor(data, dtype=torch.uint8)
data = torch.tensor(list(data), dtype=torch.uint8)
sample_count = data.shape[0] - context_length
x = data.unfold(0, context_length, 1)[:sample_count]
y = data[context_length:]