Skip to main content
torch.js has not been released yet.
torch.js logotorch.js logotorch.js
PlaygroundContact
Login
Documentation
IntroductionType SafetyTensor ExpressionsTensor IndexingEinsumEinopsAutogradTraining a ModelProfiling & MemoryPyTorch MigrationBest PracticesRuntimesPerformancePyTorch CompatibilityBenchmarksDType Coverage
IntroductionRenderer GalleryRenderersAnalysis
computeQuantilescomputePercentilescomputeCorrelationcomputeCovariancedetectOutliersdetectOutliersZScorecomputeExtendedStatsdescribeExtendedStatsQuantileResultCorrelationResultOutlierResultExtendedStatscreateDBSCANdbscanDBSCANDBSCAN.fitDBSCAN.destroyDBSCANOptionsDBSCANResultcreateHierarchicalClusteringhierarchicalClusteringHierarchicalClusteringHierarchicalClustering.fitHierarchicalClustering.destroyLinkageMethodDistanceMetricHierarchicalOptionsDendrogramNodeHierarchicalResultcomputeHistogramgetHistogramCentersnormalizeHistogramHistogramResultcreateKDEkdeKDEKDE.fitKDE.destroyKernelTypeKDEOptionsKDEResultcreateKMeanskmeansKMeansKMeans.fitKMeans.destroyKMeansOptionsKMeansResultcreatePCApcaPCAPCA.fitPCA.destroyPCAOptionsPCAResultthrottleProgressconsoleProgressProgressInfoProgressCallbackcomputeStatscomputeMinMaxdescribeStatsTensorStatscreateTSNEtsneTSNETSNE.fitTSNE.destroyTSNEOptionsTSNEResultcreateUMAPumapUMAPUMAP.fitUMAP.destroyUMAPOptionsUMAPResult
torch.js· 2026
LegalTerms of UsePrivacy Policy
/
/
  1. docs
  2. viz
  3. viz
  4. analysis
  5. UMAPOptions

viz.analysis.UMAPOptions

export interface UMAPOptions {
  /** Number of output dimensions (default: 2) */
  numDimensions?: number;
  /** Number of neighbors to consider (default: 15) */
  numNeighbors?: number;
  /** Minimum distance between points in output (default: 0.1) */
  minDist?: number;
  /** Controls how tightly points cluster (default: 1.0) */
  spread?: number;
  /** Number of optimization epochs (default: 500) */
  numEpochs?: number;
  /** Learning rate (default: 1.0) */
  learningRate?: number;
  /** Negative sample rate (default: 5) */
  negativeSampleRate?: number;
  /** Random seed for initialization */
  seed?: number;
  /** Callback for progress updates */
  onProgress?: (epoch: number, cost: number) => void;
}
numDimensions(number)optional
– Number of output dimensions (default: 2)
numNeighbors(number)optional
– Number of neighbors to consider (default: 15)
minDist(number)optional
– Minimum distance between points in output (default: 0.1)
spread(number)optional
– Controls how tightly points cluster (default: 1.0)
numEpochs(number)optional
– Number of optimization epochs (default: 500)
learningRate(number)optional
– Learning rate (default: 1.0)
negativeSampleRate(number)optional
– Negative sample rate (default: 5)
seed(number)optional
– Random seed for initialization
onProgress((epoch: number, cost: number) => void)optional
– Callback for progress updates

Options for UMAP computation.

Previous
UMAP.fit
Next
UMAPResult