From 5f06f859a53d23e081f8c9de2df5371742c6e670 Mon Sep 17 00:00:00 2001
From: "l.gabrysiak" <l.gabrysiak@osadkowski.pl>
Date: Tue, 25 Feb 2025 23:06:17 +0100
Subject: [PATCH] mod dataset

---
 hft.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/hft.py b/hft.py
index 7f4292f..7c184d5 100644
--- a/hft.py
+++ b/hft.py
@@ -207,7 +207,9 @@ def main():
         print("\nBrak danych do treningu!")
         return
 
-    dataset = Dataset.from_list(data)
+    #dataset = Dataset.from_list(data)
+    dataset = Dataset.from_dict({k: [d[k] for d in data] for k in data[0]})
+
 
     def tokenize_function(examples):
         tokenized = tokenizer(