Added optuna for experiment

myui · Dec 5, 2024 · ab6e3e3 · ab6e3e3
1 parent a7e345f
commit ab6e3e3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dockerfiles/Dockerfile-experiment b/dockerfiles/Dockerfile-experiment
@@ -53,7 +53,7 @@ RUN pip install --user --no-cache-dir --upgrade pip setuptools wheel && \
         notebook==6.5.4 ipywidgets>=8.0.6 traitlets==5.9.0 \
         widgetsnbextension pygraphviz black jupyter_contrib_nbextensions \
         matplotlib plotly tqdm memory_profiler \
-        daisyrec rectools[lightfm] implicit flurs \
+        daisyrec rectools[lightfm] implicit flurs optuna \
         # see https://github.com/lyst/lightfm/issues/709
         git+https://github.com/daviddavo/lightfm && \
     CFLAGS="-march=native" pip install git+https://github.com/tohtsky/irspack.git