test_umap / params.json
Ty-Perez's picture
UMAP fit on feed_50k: nn50 md0.2 neg10 cosine 500 epochs, no PCA
e652475 verified
Raw History Blame Contribute Delete
1.4 kB
{
"schema_version": 1,
"run_id": "20261106-090044_nn50_md0.2_cos",
"job_id": "931d1a26",
"name": "nn50 md0.2 cos",
"kind": "fit",
"status": "done",
"created_at": "2026-11-06T09:00:44-05:00",
"started_at": "2026-11-06T09:00:44-05:00",
"finished_at": "2026-11-06T09:00:50-05:00",
"feed": {
"path": "data/feed_50k.pt",
"n": 50000,
"dim": 768,
"size_bytes": 166849249,
"mtime": 1793928915.777514
},
"params": {
"n_neighbors": 50,
"min_dist": 0.2,
"spread": 1.0,
"metric": "cosine",
"random_state": null,
"use_pca": false,
"pca_n_components": 50,
"n_epochs": 500,
"learning_rate": 1.0,
"init": "spectral",
"negative_sample_rate": 10,
"repulsion_strength": 1.0,
"set_op_mix_ratio": 1.0,
"local_connectivity": 1.0,
"a": null,
"b": null,
"build_algo": "auto",
"verbose": false
},
"fitted": {
"a": 1.2620581225407888,
"b": 1.0030054002697062,
"n_epochs": 500
},
"gpu": {
"index": 2,
"name": "NVIDIA RTX A5000"
},
"timing": {
"load_feed_s": 0.999,
"import_cuml_s": 1.24,
"to_gpu_s": 0.543,
"fit_s": 1.098,
"save_model_s": 0.233,
"save_coords_s": 0.194,
"total_s": 5.43
},
"env": {
"python": "3.10.22",
"torch": "2.10.0+cu128",
"cuml": "26.02.000",
"git_commit": "159fb43"
},
"warnings": [],
"error": null
}