File tree 2 files changed +3
-1
lines changed
python/cuml/cuml/manifold
2 files changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -326,7 +326,8 @@ def simplicial_set_embedding(
326
326
327
327
cdef UMAPParams* umap_params = new UMAPParams()
328
328
umap_params.n_components = < int > n_components
329
- umap_params.initial_alpha = < int > initial_alpha
329
+ umap_params.initial_alpha = < float > initial_alpha
330
+ umap_params.learning_rate = < float > initial_alpha
330
331
umap_params.a = < float > a
331
332
umap_params.b = < float > b
332
333
Original file line number Diff line number Diff line change @@ -470,6 +470,7 @@ class UMAP(UniversalBase,
470
470
umap_params.n_components = < int > self .n_components
471
471
umap_params.n_epochs = < int > self .n_epochs if self .n_epochs else 0
472
472
umap_params.learning_rate = < float > self .learning_rate
473
+ umap_params.initial_alpha = < float > self .learning_rate
473
474
umap_params.min_dist = < float > self .min_dist
474
475
umap_params.spread = < float > self .spread
475
476
umap_params.set_op_mix_ratio = < float > self .set_op_mix_ratio
You can’t perform that action at this time.
0 commit comments