diff --git a/requirements.txt b/requirements.txt index 8913ed7..5e9f4b9 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ pandas torch -sklearn +scikit-learn torchinfo