Add data splitting and simple model

2025-03-31 15:08:39 +03:00
parent 1dc796b59e
commit f08d437b60
4 changed files with 127 additions and 2 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,4 +4,8 @@ spacy
 nltk
 matplotlib
 pandarallel
-pyarrow
+pyarrow
+scikit-learn
+torch
+transformers
+accelerate