From 2eaa18d9a156bf6f10303714c4db220faf4154a3 Mon Sep 17 00:00:00 2001 From: Marius Ciepluch <11855163+norandom@users.noreply.github.com> Date: Thu, 25 Jul 2024 16:45:57 +0200 Subject: [PATCH] added tpot deps for auto install --- dependencies/requirements.gpu.txt | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/dependencies/requirements.gpu.txt b/dependencies/requirements.gpu.txt index 1eeaea2..20dcc3e 100644 --- a/dependencies/requirements.gpu.txt +++ b/dependencies/requirements.gpu.txt @@ -14,3 +14,10 @@ sentence-transformers==2.5.1 faiss_cpu==1.8.0 linformer-pytorch==0.19.3 PyGithub==2.3.0 +deap==1.4.1 +update_checker==0.18.0 +scikit-mdr==0.4.5 +skrebate==0.62 +xgboost==2.0.3 +stopit==1.1.2 +tpot==0.12.2