diff --git a/README.md b/README.md index ee7bbbcaf..5eef0bc2e 100755 --- a/README.md +++ b/README.md @@ -113,6 +113,7 @@ The configuration of benchmarks allows you to select the frameworks to run, sele |**[PCA](https://scikit-learn.org/stable/modules/generated/sklearn.decomposition.PCA.html)**|pca|:white_check_mark:|:x:|:white_check_mark:|:white_check_mark:|:x:| |**[Ridge](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.Ridge.html)**|ridge|:white_check_mark:|:x:|:white_check_mark:|:white_check_mark:|:x:| |**[SVM](https://scikit-learn.org/stable/modules/generated/sklearn.svm.SVC.html)**|svm|:white_check_mark:|:x:|:white_check_mark:|:white_check_mark:|:x:| +|**[TSNE](https://scikit-learn.org/stable/modules/generated/sklearn.manifold.TSNE.html)**|tsne|:white_check_mark:|:x:|:x:|:white_check_mark:|:x:| |**[train_test_split](https://scikit-learn.org/stable/modules/generated/sklearn.model_selection.train_test_split.html)**|train_test_split|:white_check_mark:|:x:|:x:|:white_check_mark:|:x:| |**[GradientBoostingClassifier](https://scikit-learn.org/stable/modules/generated/sklearn.ensemble.GradientBoostingClassifier.html)**|gbt|:x:|:x:|:x:|:x:|:white_check_mark:| |**[GradientBoostingRegressor](https://scikit-learn.org/stable/modules/generated/sklearn.ensemble.GradientBoostingRegressor.html)**|gbt|:x:|:x:|:x:|:x:|:white_check_mark:| diff --git a/configs/sklearn/performance/tsne.json b/configs/sklearn/performance/tsne.json index b5d705c76..68fdc8cb7 100644 --- a/configs/sklearn/performance/tsne.json +++ b/configs/sklearn/performance/tsne.json @@ -24,15 +24,24 @@ "y": "data/mnist_y_test.npy" } }, - { - "source": "npy", - "name": "cifar_10", - "training": - { - "x": "data/cifar_10_x_train.npy", - "y": "data/cifar_10_y_train.npy" - } - } + { + "source": "npy", + "name": "cifar_10", + "training": + { + "x": "data/cifar_10_x_train.npy", + "y": "data/cifar_10_y_train.npy" + } + }, + { + "source": "npy", + "name": "epsilon_30K", + "training": + { + "x": "data/epsilon_30K_x_train.npy", + "y": "data/epsilon_30K_y_train.npy" + } + } ], "workload-size": "medium" } diff --git a/cuml_bench/README.md b/cuml_bench/README.md index e36e77f3b..edcd20735 100644 --- a/cuml_bench/README.md +++ b/cuml_bench/README.md @@ -18,6 +18,7 @@ You can launch benchmarks for each algorithm separately. The tables below list a - [PCA](#pca) - [Ridge Regression](#ridge) - [SVC](#svc) +- [TSNE](#tsne) - [train_test_split](#train_test_split) #### General @@ -141,6 +142,17 @@ You can launch benchmarks for each algorithm separately. The tables below list a | tol | float | 1e-16 | Tolerance passed to sklearn.svm.SVC | | probability | action | True | Use probability for SVC | +### TSNE + +| parameter Name | Type | default value | description | +| ----- | ---- |---- |---- | +| n-components | int | 2 | Dimension of the embedded space | +| early-exaggeration | float | 12.0 | This factor increases the attractive forces between points
and allows points to move around more freely finding their nearest neighbors more easily | +| learning-rate | float | 200.0 | The learning rate for t-SNE is usually in the range [10.0, 1000.0] | +| angle | float | 0.5 | Angular size. This is the trade-off between speed and accuracy | +| min-grad-norm | float | 1e-7 | If the gradient norm is below this threshold, the optimization is stopped | +| random-state | int | 1234 | Determines the random number generator | + #### train_test_split | parameter Name | Type | default value | description | diff --git a/sklearn_bench/README.md b/sklearn_bench/README.md index 41e0b413b..bde10c3f9 100644 --- a/sklearn_bench/README.md +++ b/sklearn_bench/README.md @@ -24,6 +24,7 @@ You can launch benchmarks for each algorithm separately. The tables below list a - [PCA](#pca) - [Ridge Regression](#ridge) - [SVC](#svc) +- [TSNE](#tsne) - [train_test_split](#train_test_split) ### General @@ -152,6 +153,17 @@ You can launch benchmarks for each algorithm separately. The tables below list a | tol | float | 1e-16 | Tolerance passed to sklearn.svm.SVC | | probability | action | True | Use probability for SVC | +### TSNE + +| parameter Name | Type | default value | description | +| ----- | ---- |---- |---- | +| n-components | int | 2 | Dimension of the embedded space | +| early-exaggeration | float | 12.0 | This factor increases the attractive forces between points
and allows points to move around more freely finding their nearest neighbors more easily | +| learning-rate | float | 200.0 | The learning rate for t-SNE is usually in the range [10.0, 1000.0] | +| angle | float | 0.5 | Angular size. This is the trade-off between speed and accuracy | +| min-grad-norm | float | 1e-7 | If the gradient norm is below this threshold, the optimization is stopped | +| random-state | int | 1234 | Determines the random number generator | + ### train_test_split | parameter Name | Type | default value | description |