Merge pull request #31 from hybrid-kg/optuna-implementation

Optuna implementation
hybrid-kg · Jul 8, 2024 · 923e8bc · 923e8bc
2 parents b0f6644 + 8609b1a
commit 923e8bc
Show file tree

Hide file tree

Showing 11 changed files with 627 additions and 513 deletions.
diff --git a/README.md b/README.md
@@ -146,10 +146,10 @@ $ clep embedding kge --data <NETWORK_FILE> --design <DESIGN_FILE> --model_config
 
 4. **Classification**
 
-The following command carries out classification on the given data file for a chosen model (Elastic Net) using a chosen optimizer (Grid Search).
+The following command carries out classification on the given data file for a chosen model (Elastic Net) with 100 hyper-parameter optimization trials.
 
 ```bash
-$ clep classify --data <EMBEDDING_FILE> --model elastic_net --optimizer grid_search --out <OUTPUT_DIR>
+$ clep classify --data <EMBEDDING_FILE> --model elastic_net --num-trials 100 --out <OUTPUT_DIR>
 ```
 
 ## Issues

diff --git a/pyproject.toml b/pyproject.toml
@@ -43,6 +43,10 @@ dependencies = [
     "scikit-optimize",
     "pykeen",
     "pandera",
+    "optuna",
+    "joblib",
+    "sqlalchemy",
+    "mysqlclient",
 ]
 
 [tool.setuptools.packages.find]

diff --git a/src/clep/classification/__init__.py b/src/clep/classification/__init__.py
@@ -2,6 +2,6 @@
 
 """Wrap Machine-Learning Classifiers for clep."""
 
-from clep.classification.classify import do_classification
+from .hpo import do_classification
 
 __all__ = ['do_classification']
diff --git a/src/clep/classification/classify.py b/src/clep/classification/classify.py