wasm-bhtsne 0.3.2 → 1.0.0

package/README.md

@@ -3,23 +3,23 @@
 <div align="center">
 
 [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
-
 </div>
 
+
 This is the wasm version of the [bhtsne](https://github.com/frjnn/bhtsne) crate.
 
-Parallel implementations of Barnes-Hut and exact implementations of the t-SNE algorithm written in Rust to run in wasm. The tree-accelerated version of the algorithm is described with fine detail in [this paper](http://lvdmaaten.github.io/publications/papers/JMLR_2014.pdf) by [Laurens van der Maaten](https://github.com/lvdmaaten). The exact, original, version of the algorithm is described in [this other paper](https://www.jmlr.org/papers/volume9/vandermaaten08a/vandermaaten08a.pdf) by [G. Hinton](https://www.cs.toronto.edu/~hinton/) and Laurens van der Maaten.
-Additional implementations of the algorithm, are listed at [this page](http://lvdmaaten.github.io/tsne/).
+## Features
+- Harnesses multi-threading capabilities through [wasm-bindgen-rayon](https://github.com/RReverser/wasm-bindgen-rayon).
 
 ## Installation
 ```shell
 npm i wasm-bhtsne
 ```
 
-### Example
+## Example
 
 ```javascript
-import { tSNE } from "wasm-bhtsne";
+import { threads } from 'wasm-feature-detect';
 
 function createRandomMatrix(rows, columns) {
     return Array.from({ length: rows }, () =>
@@ -27,15 +27,30 @@ function createRandomMatrix(rows, columns) {
     );
 }
 
-// create random points and dimensions
-const data = createRandomMatrix(5000, 512);
-
-const tsne_encoder = new tSNE(data);
-
-let compressed_vectors = []
-
-// as argument the number of epochs,
-compressed_vectors = tsne_encoder.barnes_hut(1000);
-
-console.log("Compressed Vectors:", compressed_vectors);
-```
+(async function initMultiThread() {
+    const multiThread = await import('./pkg-parallel/wasm_bhtsne.js');
+    await multiThread.default();
+    if (await threads()) {
+        console.log("Browser supports threads");
+        await multiThread.initThreadPool(navigator.hardwareConcurrency);
+    } else {
+        console.log("Browser does not support threads");
+    }
+
+    Object.assign(document.getElementById("wasm-bhtsne"), {
+        async onclick() {
+
+            // create random points and dimensions
+            const data = createRandomMatrix(500, 7);
+
+            let tsne_encoder = new multiThread.bhtSNE(data); // create a tSNE instance
+            tsne_encoder.perplexity = 25.0;  // change hyperparameters
+
+            // run the algorithm with 1000 iterations 
+            let compressed_vectors = tsne_encoder.step(1000); 
+            console.log("Compressed Vectors:", compressed_vectors);
+        },
+        disabled: false
+    });
+})();
+```

package/package.json

@@ -3,8 +3,8 @@
   "collaborators": [
     "lv291 <baiunco291@proton.me>"
   ],
-  "description": "Exact and Barnes-Hut implementations of t-SNE in wasm",
-  "version": "0.3.2",
+  "description": "Barnes-Hut implementations of t-SNE in wasm",
+  "version": "1.0.0",
   "license": "MIT",
   "repository": {
     "type": "git",
@@ -23,8 +23,8 @@
   "keywords": [
     "tsne",
     "data-visualization",
-    "data-analysis",
-    "Barnes-Hut",
-    "machine-learning"
+    "webassembly",
+    "wasm",
+    "rust"
   ]
 }

package/wasm_bhtsne.d.ts

@@ -1,52 +1,172 @@
 /* tslint:disable */
 /* eslint-disable */
 /**
+* @param {number} num_threads
+* @returns {Promise<any>}
+*/
+export function initThreadPool(num_threads: number): Promise<any>;
+/**
+* @param {number} receiver
+*/
+export function wbg_rayon_start_worker(receiver: number): void;
+/**
 * t-distributed stochastic neighbor embedding. Provides a parallel implementation of both the
 * exact version of the algorithm and the tree accelerated one leveraging space partitioning trees.
 */
-export class tSNE {
+export class bhtSNE {
   free(): void;
 /**
-* @param {Array<any>} vectors
+* @param {any} data
 */
-  constructor(vectors: Array<any>);
+  constructor(data: any);
 /**
 * Performs a parallel Barnes-Hut approximation of the t-SNE algorithm.
 *
 * # Arguments
 *
-* * `theta` - determines the accuracy of the approximation. Must be **strictly greater than
+* `epochs` - Sets epochs, the maximum number of fitting iterations.
+* @param {number} epochs
+* @returns {any}
+*/
+  step(epochs: number): any;
+/**
+* Sets a new learning rate.
+*
+* # Arguments
+*
+* `learning_rate` - new value for the learning rate.
+* @param {number} learning_rate
+*/
+  learning_rate(learning_rate: number): void;
+/**
+* Sets new epochs, i.e the maximum number of fitting iterations.
+*
+* # Arguments
+*
+* `epochs` - new value for the epochs.
+* @param {number} epochs
+*/
+  epochs(epochs: number): void;
+/**
+* Sets a new momentum.
+*
+* # Arguments
+*
+* `momentum` - new value for the momentum.
+* @param {number} momentum
+*/
+  momentum(momentum: number): void;
+/**
+* Sets a new final momentum.
+*
+* # Arguments
+*
+* `final_momentum` - new value for the final momentum.
+* @param {number} final_momentum
+*/
+  final_momentum(final_momentum: number): void;
+/**
+* Sets a new momentum switch epoch, i.e. the epoch after which the algorithm switches to
+* `final_momentum` for the map update.
+*
+* # Arguments
+*
+* `momentum_switch_epoch` - new value for the momentum switch epoch.
+* @param {number} momentum_switch_epoch
+*/
+  momentum_switch_epoch(momentum_switch_epoch: number): void;
+/**
+* Sets a new stop lying epoch, i.e. the epoch after which the P distribution values become
+* true, as defined in the original implementation. For epochs < `stop_lying_epoch` the values
+* of the P distribution are multiplied by a factor equal to `12.0`.
+*
+* # Arguments
+*
+* `stop_lying_epoch` - new value for the stop lying epoch.
+* @param {number} stop_lying_epoch
+*/
+  stop_lying_epoch(stop_lying_epoch: number): void;
+/**
+* Sets a new theta, which determines the accuracy of the approximation. Must be **strictly greater than
 * 0.0**. Large values for θ increase the speed of the algorithm but decrease its accuracy.
 * For small values of θ it is less probable that a cell in the space partitioning tree will
 * be treated as a single point. For θ equal to 0.0 the method degenerates in the exact
 * version.
 *
-* * `metric_f` - metric function.
+* # Arguments
 *
+* * `theta`  - new value for the theta.
+* @param {number} theta
+*/
+  theta(theta: number): void;
+/**
+* Sets a new value for the embedding dimension.
 *
-* **Do note that** `metric_f` **must be a metric distance**, i.e. it must
-* satisfy the [triangle inequality](https://en.wikipedia.org/wiki/Triangle_inequality).
-* @param {number} epochs
-* @returns {Array<any>}
+* # Arguments
+*
+* `embedding_dim` - new value for the embedding space dimensionality.
+* @param {number} embedding_dim
+*/
+  embedding_dim(embedding_dim: number): void;
+/**
+* Sets a new perplexity value.
+*
+* # Arguments
+*
+* `perplexity` - new value for the perplexity. It's used so that the bandwidth of the Gaussian
+*  kernels, is set in such a way that the perplexity of each the conditional distribution *Pi*
+*  equals a predefined perplexity *u*.
+*
+* A good value for perplexity lies between 5.0 and 50.0.
+* @param {number} perplexity
+*/
+  perplexity(perplexity: number): void;
+}
+/**
+*/
+export class wbg_rayon_PoolBuilder {
+  free(): void;
+/**
+* @returns {number}
+*/
+  numThreads(): number;
+/**
+* @returns {number}
 */
-  barnes_hut(epochs: number): Array<any>;
+  receiver(): number;
 /**
 */
-  theta: number;
+  build(): void;
 }
 
 export type InitInput = RequestInfo | URL | Response | BufferSource | WebAssembly.Module;
 
 export interface InitOutput {
+  readonly __wbg_bhtsne_free: (a: number) => void;
+  readonly bhtsne_new: (a: number) => number;
+  readonly bhtsne_step: (a: number, b: number, c: number) => void;
+  readonly bhtsne_learning_rate: (a: number, b: number) => void;
+  readonly bhtsne_epochs: (a: number, b: number) => void;
+  readonly bhtsne_momentum: (a: number, b: number) => void;
+  readonly bhtsne_final_momentum: (a: number, b: number) => void;
+  readonly bhtsne_momentum_switch_epoch: (a: number, b: number) => void;
+  readonly bhtsne_stop_lying_epoch: (a: number, b: number) => void;
+  readonly bhtsne_theta: (a: number, b: number) => void;
+  readonly bhtsne_embedding_dim: (a: number, b: number) => void;
+  readonly bhtsne_perplexity: (a: number, b: number) => void;
+  readonly __wbg_wbg_rayon_poolbuilder_free: (a: number) => void;
+  readonly wbg_rayon_poolbuilder_numThreads: (a: number) => number;
+  readonly wbg_rayon_poolbuilder_receiver: (a: number) => number;
+  readonly wbg_rayon_poolbuilder_build: (a: number) => void;
+  readonly initThreadPool: (a: number) => number;
+  readonly wbg_rayon_start_worker: (a: number) => void;
   readonly memory: WebAssembly.Memory;
-  readonly __wbg_tsne_free: (a: number) => void;
-  readonly tsne_new: (a: number) => number;
-  readonly tsne_barnes_hut: (a: number, b: number) => number;
-  readonly tsne_set_theta: (a: number, b: number) => void;
   readonly __wbindgen_malloc: (a: number, b: number) => number;
   readonly __wbindgen_realloc: (a: number, b: number, c: number, d: number) => number;
-  readonly __wbindgen_free: (a: number, b: number, c: number) => void;
+  readonly __wbindgen_add_to_stack_pointer: (a: number) => number;
   readonly __wbindgen_exn_store: (a: number) => void;
+  readonly __wbindgen_thread_destroy: (a?: number, b?: number) => void;
+  readonly __wbindgen_start: () => void;
 }
 
 export type SyncInitInput = BufferSource | WebAssembly.Module;
@@ -55,17 +175,19 @@ export type SyncInitInput = BufferSource | WebAssembly.Module;
 * a precompiled `WebAssembly.Module`.
 *
 * @param {SyncInitInput} module
+* @param {WebAssembly.Memory} maybe_memory
 *
 * @returns {InitOutput}
 */
-export function initSync(module: SyncInitInput): InitOutput;
+export function initSync(module: SyncInitInput, maybe_memory?: WebAssembly.Memory): InitOutput;
 
 /**
 * If `module_or_path` is {RequestInfo} or {URL}, makes a request and
 * for everything else, calls `WebAssembly.instantiate` directly.
 *
 * @param {InitInput | Promise<InitInput>} module_or_path
+* @param {WebAssembly.Memory} maybe_memory
 *
 * @returns {Promise<InitOutput>}
 */
-export default function __wbg_init (module_or_path?: InitInput | Promise<InitInput>): Promise<InitOutput>;
+export default function __wbg_init (module_or_path?: InitInput | Promise<InitInput>, maybe_memory?: WebAssembly.Memory): Promise<InitOutput>;

package/wasm_bhtsne.js

@@ -1,3 +1,5 @@
+import { startWorkers } from './snippets/wasm-bindgen-rayon-3e04391371ad0a8e/src/workerHelpers.js';
+
 let wasm;
 
 const heap = new Array(128).fill(undefined);
@@ -27,7 +29,7 @@ function isLikeNone(x) {
 let cachedFloat64Memory0 = null;
 
 function getFloat64Memory0() {
-    if (cachedFloat64Memory0 === null || cachedFloat64Memory0.byteLength === 0) {
+    if (cachedFloat64Memory0 === null || cachedFloat64Memory0.buffer !== wasm.memory.buffer) {
         cachedFloat64Memory0 = new Float64Array(wasm.memory.buffer);
     }
     return cachedFloat64Memory0;
@@ -36,37 +38,89 @@ function getFloat64Memory0() {
 let cachedInt32Memory0 = null;
 
 function getInt32Memory0() {
-    if (cachedInt32Memory0 === null || cachedInt32Memory0.byteLength === 0) {
+    if (cachedInt32Memory0 === null || cachedInt32Memory0.buffer !== wasm.memory.buffer) {
         cachedInt32Memory0 = new Int32Array(wasm.memory.buffer);
     }
     return cachedInt32Memory0;
 }
 
-const cachedTextDecoder = (typeof TextDecoder !== 'undefined' ? new TextDecoder('utf-8', { ignoreBOM: true, fatal: true }) : { decode: () => { throw Error('TextDecoder not available') } } );
+function addHeapObject(obj) {
+    if (heap_next === heap.length) heap.push(heap.length + 1);
+    const idx = heap_next;
+    heap_next = heap[idx];
 
-if (typeof TextDecoder !== 'undefined') { cachedTextDecoder.decode(); };
+    heap[idx] = obj;
+    return idx;
+}
+
+let WASM_VECTOR_LEN = 0;
 
 let cachedUint8Memory0 = null;
 
 function getUint8Memory0() {
-    if (cachedUint8Memory0 === null || cachedUint8Memory0.byteLength === 0) {
+    if (cachedUint8Memory0 === null || cachedUint8Memory0.buffer !== wasm.memory.buffer) {
         cachedUint8Memory0 = new Uint8Array(wasm.memory.buffer);
     }
     return cachedUint8Memory0;
 }
 
-function getStringFromWasm0(ptr, len) {
-    ptr = ptr >>> 0;
-    return cachedTextDecoder.decode(getUint8Memory0().subarray(ptr, ptr + len));
+const cachedTextEncoder = (typeof TextEncoder !== 'undefined' ? new TextEncoder('utf-8') : { encode: () => { throw Error('TextEncoder not available') } } );
+
+const encodeString = function (arg, view) {
+    const buf = cachedTextEncoder.encode(arg);
+    view.set(buf);
+    return {
+        read: arg.length,
+        written: buf.length
+    };
+};
+
+function passStringToWasm0(arg, malloc, realloc) {
+
+    if (realloc === undefined) {
+        const buf = cachedTextEncoder.encode(arg);
+        const ptr = malloc(buf.length, 1) >>> 0;
+        getUint8Memory0().subarray(ptr, ptr + buf.length).set(buf);
+        WASM_VECTOR_LEN = buf.length;
+        return ptr;
+    }
+
+    let len = arg.length;
+    let ptr = malloc(len, 1) >>> 0;
+
+    const mem = getUint8Memory0();
+
+    let offset = 0;
+
+    for (; offset < len; offset++) {
+        const code = arg.charCodeAt(offset);
+        if (code > 0x7F) break;
+        mem[ptr + offset] = code;
+    }
+
+    if (offset !== len) {
+        if (offset !== 0) {
+            arg = arg.slice(offset);
+        }
+        ptr = realloc(ptr, len, len = offset + arg.length * 3, 1) >>> 0;
+        const view = getUint8Memory0().subarray(ptr + offset, ptr + len);
+        const ret = encodeString(arg, view);
+
+        offset += ret.written;
+        ptr = realloc(ptr, len, offset, 1) >>> 0;
+    }
+
+    WASM_VECTOR_LEN = offset;
+    return ptr;
 }
 
-function addHeapObject(obj) {
-    if (heap_next === heap.length) heap.push(heap.length + 1);
-    const idx = heap_next;
-    heap_next = heap[idx];
+const cachedTextDecoder = (typeof TextDecoder !== 'undefined' ? new TextDecoder('utf-8', { ignoreBOM: true, fatal: true }) : { decode: () => { throw Error('TextDecoder not available') } } );
 
-    heap[idx] = obj;
-    return idx;
+if (typeof TextDecoder !== 'undefined') { cachedTextDecoder.decode(); };
+
+function getStringFromWasm0(ptr, len) {
+    ptr = ptr >>> 0;
+    return cachedTextDecoder.decode(getUint8Memory0().slice(ptr, ptr + len));
 }
 
 function debugString(val) {
@@ -134,61 +188,6 @@ function debugString(val) {
     return className;
 }
 
-let WASM_VECTOR_LEN = 0;
-
-const cachedTextEncoder = (typeof TextEncoder !== 'undefined' ? new TextEncoder('utf-8') : { encode: () => { throw Error('TextEncoder not available') } } );
-
-const encodeString = (typeof cachedTextEncoder.encodeInto === 'function'
-    ? function (arg, view) {
-    return cachedTextEncoder.encodeInto(arg, view);
-}
-    : function (arg, view) {
-    const buf = cachedTextEncoder.encode(arg);
-    view.set(buf);
-    return {
-        read: arg.length,
-        written: buf.length
-    };
-});
-
-function passStringToWasm0(arg, malloc, realloc) {
-
-    if (realloc === undefined) {
-        const buf = cachedTextEncoder.encode(arg);
-        const ptr = malloc(buf.length, 1) >>> 0;
-        getUint8Memory0().subarray(ptr, ptr + buf.length).set(buf);
-        WASM_VECTOR_LEN = buf.length;
-        return ptr;
-    }
-
-    let len = arg.length;
-    let ptr = malloc(len, 1) >>> 0;
-
-    const mem = getUint8Memory0();
-
-    let offset = 0;
-
-    for (; offset < len; offset++) {
-        const code = arg.charCodeAt(offset);
-        if (code > 0x7F) break;
-        mem[ptr + offset] = code;
-    }
-
-    if (offset !== len) {
-        if (offset !== 0) {
-            arg = arg.slice(offset);
-        }
-        ptr = realloc(ptr, len, len = offset + arg.length * 3, 1) >>> 0;
-        const view = getUint8Memory0().subarray(ptr + offset, ptr + len);
-        const ret = encodeString(arg, view);
-
-        offset += ret.written;
-    }
-
-    WASM_VECTOR_LEN = offset;
-    return ptr;
-}
-
 function handleError(f, args) {
     try {
         return f.apply(this, args);
@@ -197,65 +196,229 @@ function handleError(f, args) {
     }
 }
 /**
-* t-distributed stochastic neighbor embedding. Provides a parallel implementation of both the
-* exact version of the algorithm and the tree accelerated one leveraging space partitioning trees.
+* @param {number} num_threads
+* @returns {Promise<any>}
 */
-export class tSNE {
+export function initThreadPool(num_threads) {
+    const ret = wasm.initThreadPool(num_threads);
+    return takeObject(ret);
+}
 
-    static __wrap(ptr) {
-        ptr = ptr >>> 0;
-        const obj = Object.create(tSNE.prototype);
-        obj.__wbg_ptr = ptr;
+/**
+* @param {number} receiver
+*/
+export function wbg_rayon_start_worker(receiver) {
+    wasm.wbg_rayon_start_worker(receiver);
+}
 
-        return obj;
-    }
+const bhtSNEFinalization = (typeof FinalizationRegistry === 'undefined')
+    ? { register: () => {}, unregister: () => {} }
+    : new FinalizationRegistry(ptr => wasm.__wbg_bhtsne_free(ptr >>> 0));
+/**
+* t-distributed stochastic neighbor embedding. Provides a parallel implementation of both the
+* exact version of the algorithm and the tree accelerated one leveraging space partitioning trees.
+*/
+export class bhtSNE {
 
     __destroy_into_raw() {
         const ptr = this.__wbg_ptr;
         this.__wbg_ptr = 0;
-
+        bhtSNEFinalization.unregister(this);
         return ptr;
     }
 
     free() {
         const ptr = this.__destroy_into_raw();
-        wasm.__wbg_tsne_free(ptr);
+        wasm.__wbg_bhtsne_free(ptr);
     }
     /**
-    * @param {Array<any>} vectors
+    * @param {any} data
     */
-    constructor(vectors) {
-        const ret = wasm.tsne_new(addHeapObject(vectors));
-        return tSNE.__wrap(ret);
+    constructor(data) {
+        const ret = wasm.bhtsne_new(addHeapObject(data));
+        this.__wbg_ptr = ret >>> 0;
+        return this;
     }
     /**
     * Performs a parallel Barnes-Hut approximation of the t-SNE algorithm.
     *
     * # Arguments
     *
-    * * `theta` - determines the accuracy of the approximation. Must be **strictly greater than
+    * `epochs` - Sets epochs, the maximum number of fitting iterations.
+    * @param {number} epochs
+    * @returns {any}
+    */
+    step(epochs) {
+        try {
+            const retptr = wasm.__wbindgen_add_to_stack_pointer(-16);
+            wasm.bhtsne_step(retptr, this.__wbg_ptr, epochs);
+            var r0 = getInt32Memory0()[retptr / 4 + 0];
+            var r1 = getInt32Memory0()[retptr / 4 + 1];
+            var r2 = getInt32Memory0()[retptr / 4 + 2];
+            if (r2) {
+                throw takeObject(r1);
+            }
+            return takeObject(r0);
+        } finally {
+            wasm.__wbindgen_add_to_stack_pointer(16);
+        }
+    }
+    /**
+    * Sets a new learning rate.
+    *
+    * # Arguments
+    *
+    * `learning_rate` - new value for the learning rate.
+    * @param {number} learning_rate
+    */
+    learning_rate(learning_rate) {
+        wasm.bhtsne_learning_rate(this.__wbg_ptr, learning_rate);
+    }
+    /**
+    * Sets new epochs, i.e the maximum number of fitting iterations.
+    *
+    * # Arguments
+    *
+    * `epochs` - new value for the epochs.
+    * @param {number} epochs
+    */
+    epochs(epochs) {
+        wasm.bhtsne_epochs(this.__wbg_ptr, epochs);
+    }
+    /**
+    * Sets a new momentum.
+    *
+    * # Arguments
+    *
+    * `momentum` - new value for the momentum.
+    * @param {number} momentum
+    */
+    momentum(momentum) {
+        wasm.bhtsne_momentum(this.__wbg_ptr, momentum);
+    }
+    /**
+    * Sets a new final momentum.
+    *
+    * # Arguments
+    *
+    * `final_momentum` - new value for the final momentum.
+    * @param {number} final_momentum
+    */
+    final_momentum(final_momentum) {
+        wasm.bhtsne_final_momentum(this.__wbg_ptr, final_momentum);
+    }
+    /**
+    * Sets a new momentum switch epoch, i.e. the epoch after which the algorithm switches to
+    * `final_momentum` for the map update.
+    *
+    * # Arguments
+    *
+    * `momentum_switch_epoch` - new value for the momentum switch epoch.
+    * @param {number} momentum_switch_epoch
+    */
+    momentum_switch_epoch(momentum_switch_epoch) {
+        wasm.bhtsne_momentum_switch_epoch(this.__wbg_ptr, momentum_switch_epoch);
+    }
+    /**
+    * Sets a new stop lying epoch, i.e. the epoch after which the P distribution values become
+    * true, as defined in the original implementation. For epochs < `stop_lying_epoch` the values
+    * of the P distribution are multiplied by a factor equal to `12.0`.
+    *
+    * # Arguments
+    *
+    * `stop_lying_epoch` - new value for the stop lying epoch.
+    * @param {number} stop_lying_epoch
+    */
+    stop_lying_epoch(stop_lying_epoch) {
+        wasm.bhtsne_stop_lying_epoch(this.__wbg_ptr, stop_lying_epoch);
+    }
+    /**
+    * Sets a new theta, which determines the accuracy of the approximation. Must be **strictly greater than
     * 0.0**. Large values for θ increase the speed of the algorithm but decrease its accuracy.
     * For small values of θ it is less probable that a cell in the space partitioning tree will
     * be treated as a single point. For θ equal to 0.0 the method degenerates in the exact
     * version.
     *
-    * * `metric_f` - metric function.
+    * # Arguments
     *
+    * * `theta`  - new value for the theta.
+    * @param {number} theta
+    */
+    theta(theta) {
+        wasm.bhtsne_theta(this.__wbg_ptr, theta);
+    }
+    /**
+    * Sets a new value for the embedding dimension.
     *
-    * **Do note that** `metric_f` **must be a metric distance**, i.e. it must
-    * satisfy the [triangle inequality](https://en.wikipedia.org/wiki/Triangle_inequality).
-    * @param {number} epochs
-    * @returns {Array<any>}
+    * # Arguments
+    *
+    * `embedding_dim` - new value for the embedding space dimensionality.
+    * @param {number} embedding_dim
     */
-    barnes_hut(epochs) {
-        const ret = wasm.tsne_barnes_hut(this.__wbg_ptr, epochs);
-        return takeObject(ret);
+    embedding_dim(embedding_dim) {
+        wasm.bhtsne_embedding_dim(this.__wbg_ptr, embedding_dim);
     }
     /**
-    * @param {number} theta
+    * Sets a new perplexity value.
+    *
+    * # Arguments
+    *
+    * `perplexity` - new value for the perplexity. It's used so that the bandwidth of the Gaussian
+    *  kernels, is set in such a way that the perplexity of each the conditional distribution *Pi*
+    *  equals a predefined perplexity *u*.
+    *
+    * A good value for perplexity lies between 5.0 and 50.0.
+    * @param {number} perplexity
+    */
+    perplexity(perplexity) {
+        wasm.bhtsne_perplexity(this.__wbg_ptr, perplexity);
+    }
+}
+
+const wbg_rayon_PoolBuilderFinalization = (typeof FinalizationRegistry === 'undefined')
+    ? { register: () => {}, unregister: () => {} }
+    : new FinalizationRegistry(ptr => wasm.__wbg_wbg_rayon_poolbuilder_free(ptr >>> 0));
+/**
+*/
+export class wbg_rayon_PoolBuilder {
+
+    static __wrap(ptr) {
+        ptr = ptr >>> 0;
+        const obj = Object.create(wbg_rayon_PoolBuilder.prototype);
+        obj.__wbg_ptr = ptr;
+        wbg_rayon_PoolBuilderFinalization.register(obj, obj.__wbg_ptr, obj);
+        return obj;
+    }
+
+    __destroy_into_raw() {
+        const ptr = this.__wbg_ptr;
+        this.__wbg_ptr = 0;
+        wbg_rayon_PoolBuilderFinalization.unregister(this);
+        return ptr;
+    }
+
+    free() {
+        const ptr = this.__destroy_into_raw();
+        wasm.__wbg_wbg_rayon_poolbuilder_free(ptr);
+    }
+    /**
+    * @returns {number}
     */
-    set theta(theta) {
-        wasm.tsne_set_theta(this.__wbg_ptr, theta);
+    numThreads() {
+        const ret = wasm.wbg_rayon_poolbuilder_numThreads(this.__wbg_ptr);
+        return ret >>> 0;
+    }
+    /**
+    * @returns {number}
+    */
+    receiver() {
+        const ret = wasm.wbg_rayon_poolbuilder_receiver(this.__wbg_ptr);
+        return ret >>> 0;
+    }
+    /**
+    */
+    build() {
+        wasm.wbg_rayon_poolbuilder_build(this.__wbg_ptr);
     }
 }
 
@@ -302,52 +465,57 @@ function __wbg_get_imports() {
         getFloat64Memory0()[arg0 / 8 + 1] = isLikeNone(ret) ? 0 : ret;
         getInt32Memory0()[arg0 / 4 + 0] = !isLikeNone(ret);
     };
-    imports.wbg.__wbg_new_abda76e883ba8a5f = function() {
-        const ret = new Error();
+    imports.wbg.__wbindgen_object_clone_ref = function(arg0) {
+        const ret = getObject(arg0);
         return addHeapObject(ret);
     };
-    imports.wbg.__wbg_stack_658279fe44541cf6 = function(arg0, arg1) {
-        const ret = getObject(arg1).stack;
-        const ptr1 = passStringToWasm0(ret, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
-        const len1 = WASM_VECTOR_LEN;
+    imports.wbg.__wbindgen_is_object = function(arg0) {
+        const val = getObject(arg0);
+        const ret = typeof(val) === 'object' && val !== null;
+        return ret;
+    };
+    imports.wbg.__wbindgen_jsval_loose_eq = function(arg0, arg1) {
+        const ret = getObject(arg0) == getObject(arg1);
+        return ret;
+    };
+    imports.wbg.__wbindgen_boolean_get = function(arg0) {
+        const v = getObject(arg0);
+        const ret = typeof(v) === 'boolean' ? (v ? 1 : 0) : 2;
+        return ret;
+    };
+    imports.wbg.__wbindgen_string_get = function(arg0, arg1) {
+        const obj = getObject(arg1);
+        const ret = typeof(obj) === 'string' ? obj : undefined;
+        var ptr1 = isLikeNone(ret) ? 0 : passStringToWasm0(ret, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        var len1 = WASM_VECTOR_LEN;
         getInt32Memory0()[arg0 / 4 + 1] = len1;
         getInt32Memory0()[arg0 / 4 + 0] = ptr1;
     };
-    imports.wbg.__wbg_error_f851667af71bcfc6 = function(arg0, arg1) {
-        let deferred0_0;
-        let deferred0_1;
-        try {
-            deferred0_0 = arg0;
-            deferred0_1 = arg1;
-            console.error(getStringFromWasm0(arg0, arg1));
-        } finally {
-            wasm.__wbindgen_free(deferred0_0, deferred0_1, 1);
-        }
+    imports.wbg.__wbindgen_error_new = function(arg0, arg1) {
+        const ret = new Error(getStringFromWasm0(arg0, arg1));
+        return addHeapObject(ret);
     };
-    imports.wbg.__wbg_randomFillSync_dc1e9a60c158336d = function() { return handleError(function (arg0, arg1) {
-        getObject(arg0).randomFillSync(takeObject(arg1));
-    }, arguments) };
-    imports.wbg.__wbg_getRandomValues_37fa2ca9e4e07fab = function() { return handleError(function (arg0, arg1) {
-        getObject(arg0).getRandomValues(getObject(arg1));
-    }, arguments) };
-    imports.wbg.__wbg_crypto_c48a774b022d20ac = function(arg0) {
-        const ret = getObject(arg0).crypto;
+    imports.wbg.__wbindgen_number_new = function(arg0) {
+        const ret = arg0;
         return addHeapObject(ret);
     };
-    imports.wbg.__wbindgen_is_object = function(arg0) {
-        const val = getObject(arg0);
-        const ret = typeof(val) === 'object' && val !== null;
-        return ret;
+    imports.wbg.__wbindgen_string_new = function(arg0, arg1) {
+        const ret = getStringFromWasm0(arg0, arg1);
+        return addHeapObject(ret);
     };
-    imports.wbg.__wbg_process_298734cf255a885d = function(arg0) {
+    imports.wbg.__wbg_crypto_1d1f22824a6a080c = function(arg0) {
+        const ret = getObject(arg0).crypto;
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbg_process_4a72847cc503995b = function(arg0) {
         const ret = getObject(arg0).process;
         return addHeapObject(ret);
     };
-    imports.wbg.__wbg_versions_e2e78e134e3e5d01 = function(arg0) {
+    imports.wbg.__wbg_versions_f686565e586dd935 = function(arg0) {
         const ret = getObject(arg0).versions;
         return addHeapObject(ret);
     };
-    imports.wbg.__wbg_node_1cd7a5d853dbea79 = function(arg0) {
+    imports.wbg.__wbg_node_104a2ff8d6ea03a2 = function(arg0) {
         const ret = getObject(arg0).node;
         return addHeapObject(ret);
     };
@@ -355,27 +523,39 @@ function __wbg_get_imports() {
         const ret = typeof(getObject(arg0)) === 'string';
         return ret;
     };
-    imports.wbg.__wbg_require_8f08ceecec0f4fee = function() { return handleError(function () {
+    imports.wbg.__wbg_require_cca90b1a94a0255b = function() { return handleError(function () {
         const ret = module.require;
         return addHeapObject(ret);
     }, arguments) };
-    imports.wbg.__wbindgen_string_new = function(arg0, arg1) {
-        const ret = getStringFromWasm0(arg0, arg1);
-        return addHeapObject(ret);
-    };
-    imports.wbg.__wbg_msCrypto_bcb970640f50a1e8 = function(arg0) {
+    imports.wbg.__wbg_msCrypto_eb05e62b530a1508 = function(arg0) {
         const ret = getObject(arg0).msCrypto;
         return addHeapObject(ret);
     };
-    imports.wbg.__wbg_get_44be0491f933a435 = function(arg0, arg1) {
+    imports.wbg.__wbg_randomFillSync_5c9c955aa56b6049 = function() { return handleError(function (arg0, arg1) {
+        getObject(arg0).randomFillSync(takeObject(arg1));
+    }, arguments) };
+    imports.wbg.__wbg_getRandomValues_3aa56aa6edec874c = function() { return handleError(function (arg0, arg1) {
+        getObject(arg0).getRandomValues(getObject(arg1));
+    }, arguments) };
+    imports.wbg.__wbg_instanceof_Window_f401953a2cf86220 = function(arg0) {
+        let result;
+        try {
+            result = getObject(arg0) instanceof Window;
+        } catch (_) {
+            result = false;
+        }
+        const ret = result;
+        return ret;
+    };
+    imports.wbg.__wbg_get_bd8e338fbd5f5cc8 = function(arg0, arg1) {
         const ret = getObject(arg0)[arg1 >>> 0];
         return addHeapObject(ret);
     };
-    imports.wbg.__wbg_length_fff51ee6522a1a18 = function(arg0) {
+    imports.wbg.__wbg_length_cd7af8117672b8b8 = function(arg0) {
         const ret = getObject(arg0).length;
         return ret;
     };
-    imports.wbg.__wbg_new_898a68150f225f2e = function() {
+    imports.wbg.__wbg_new_16b304a2cfa7ff4a = function() {
         const ret = new Array();
         return addHeapObject(ret);
     };
@@ -383,27 +563,51 @@ function __wbg_get_imports() {
         const ret = typeof(getObject(arg0)) === 'function';
         return ret;
     };
-    imports.wbg.__wbg_newnoargs_581967eacc0e2604 = function(arg0, arg1) {
+    imports.wbg.__wbg_newnoargs_e258087cd0daa0ea = function(arg0, arg1) {
         const ret = new Function(getStringFromWasm0(arg0, arg1));
         return addHeapObject(ret);
     };
-    imports.wbg.__wbg_call_cb65541d95d71282 = function() { return handleError(function (arg0, arg1) {
+    imports.wbg.__wbg_next_40fc327bfc8770e6 = function(arg0) {
+        const ret = getObject(arg0).next;
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbg_next_196c84450b364254 = function() { return handleError(function (arg0) {
+        const ret = getObject(arg0).next();
+        return addHeapObject(ret);
+    }, arguments) };
+    imports.wbg.__wbg_done_298b57d23c0fc80c = function(arg0) {
+        const ret = getObject(arg0).done;
+        return ret;
+    };
+    imports.wbg.__wbg_value_d93c65011f51a456 = function(arg0) {
+        const ret = getObject(arg0).value;
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbg_iterator_2cee6dadfd956dfa = function() {
+        const ret = Symbol.iterator;
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbg_get_e3c254076557e348 = function() { return handleError(function (arg0, arg1) {
+        const ret = Reflect.get(getObject(arg0), getObject(arg1));
+        return addHeapObject(ret);
+    }, arguments) };
+    imports.wbg.__wbg_call_27c0f87801dedf93 = function() { return handleError(function (arg0, arg1) {
         const ret = getObject(arg0).call(getObject(arg1));
         return addHeapObject(ret);
     }, arguments) };
-    imports.wbg.__wbg_self_1ff1d729e9aae938 = function() { return handleError(function () {
+    imports.wbg.__wbg_self_ce0dbfc45cf2f5be = function() { return handleError(function () {
         const ret = self.self;
         return addHeapObject(ret);
     }, arguments) };
-    imports.wbg.__wbg_window_5f4faef6c12b79ec = function() { return handleError(function () {
+    imports.wbg.__wbg_window_c6fb939a7f436783 = function() { return handleError(function () {
         const ret = window.window;
         return addHeapObject(ret);
     }, arguments) };
-    imports.wbg.__wbg_globalThis_1d39714405582d3c = function() { return handleError(function () {
+    imports.wbg.__wbg_globalThis_d1e6af4856ba331b = function() { return handleError(function () {
         const ret = globalThis.globalThis;
         return addHeapObject(ret);
     }, arguments) };
-    imports.wbg.__wbg_global_651f05c6a0944d1c = function() { return handleError(function () {
+    imports.wbg.__wbg_global_207b558942527489 = function() { return handleError(function () {
         const ret = global.global;
         return addHeapObject(ret);
     }, arguments) };
@@ -411,53 +615,64 @@ function __wbg_get_imports() {
         const ret = getObject(arg0) === undefined;
         return ret;
     };
-    imports.wbg.__wbg_isArray_4c24b343cb13cfb1 = function(arg0) {
+    imports.wbg.__wbg_set_d4638f722068f043 = function(arg0, arg1, arg2) {
+        getObject(arg0)[arg1 >>> 0] = takeObject(arg2);
+    };
+    imports.wbg.__wbg_isArray_2ab64d95e09ea0ae = function(arg0) {
         const ret = Array.isArray(getObject(arg0));
         return ret;
     };
-    imports.wbg.__wbg_push_ca1c26067ef907ac = function(arg0, arg1) {
-        const ret = getObject(arg0).push(getObject(arg1));
+    imports.wbg.__wbg_instanceof_ArrayBuffer_836825be07d4c9d2 = function(arg0) {
+        let result;
+        try {
+            result = getObject(arg0) instanceof ArrayBuffer;
+        } catch (_) {
+            result = false;
+        }
+        const ret = result;
         return ret;
     };
-    imports.wbg.__wbg_call_01734de55d61e11d = function() { return handleError(function (arg0, arg1, arg2) {
+    imports.wbg.__wbg_call_b3ca7c6051f9bec1 = function() { return handleError(function (arg0, arg1, arg2) {
         const ret = getObject(arg0).call(getObject(arg1), getObject(arg2));
         return addHeapObject(ret);
     }, arguments) };
-    imports.wbg.__wbg_buffer_085ec1f694018c4f = function(arg0) {
+    imports.wbg.__wbg_buffer_12d079cc21e14bdb = function(arg0) {
         const ret = getObject(arg0).buffer;
         return addHeapObject(ret);
     };
-    imports.wbg.__wbg_newwithbyteoffsetandlength_6da8e527659b86aa = function(arg0, arg1, arg2) {
+    imports.wbg.__wbg_newwithbyteoffsetandlength_aa4a17c33a06e5cb = function(arg0, arg1, arg2) {
         const ret = new Uint8Array(getObject(arg0), arg1 >>> 0, arg2 >>> 0);
         return addHeapObject(ret);
     };
-    imports.wbg.__wbg_new_8125e318e6245eed = function(arg0) {
+    imports.wbg.__wbg_new_63b92bc8671ed464 = function(arg0) {
         const ret = new Uint8Array(getObject(arg0));
         return addHeapObject(ret);
     };
-    imports.wbg.__wbg_set_5cf90238115182c3 = function(arg0, arg1, arg2) {
+    imports.wbg.__wbg_set_a47bac70306a19a7 = function(arg0, arg1, arg2) {
         getObject(arg0).set(getObject(arg1), arg2 >>> 0);
     };
-    imports.wbg.__wbg_newwithbyteoffsetandlength_69193e31c844b792 = function(arg0, arg1, arg2) {
-        const ret = new Float32Array(getObject(arg0), arg1 >>> 0, arg2 >>> 0);
-        return addHeapObject(ret);
+    imports.wbg.__wbg_length_c20a40f15020d68a = function(arg0) {
+        const ret = getObject(arg0).length;
+        return ret;
     };
-    imports.wbg.__wbg_new_d086a66d1c264b3f = function(arg0) {
-        const ret = new Float32Array(getObject(arg0));
-        return addHeapObject(ret);
+    imports.wbg.__wbg_instanceof_Uint8Array_2b3bbecd033d19f6 = function(arg0) {
+        let result;
+        try {
+            result = getObject(arg0) instanceof Uint8Array;
+        } catch (_) {
+            result = false;
+        }
+        const ret = result;
+        return ret;
     };
-    imports.wbg.__wbg_newwithlength_e5d69174d6984cd7 = function(arg0) {
+    imports.wbg.__wbg_newwithlength_e9b4878cebadb3d3 = function(arg0) {
         const ret = new Uint8Array(arg0 >>> 0);
         return addHeapObject(ret);
     };
-    imports.wbg.__wbg_subarray_13db269f57aa838d = function(arg0, arg1, arg2) {
+    imports.wbg.__wbg_subarray_a1f73cd4b5b42fe1 = function(arg0, arg1, arg2) {
         const ret = getObject(arg0).subarray(arg1 >>> 0, arg2 >>> 0);
         return addHeapObject(ret);
     };
-    imports.wbg.__wbindgen_object_clone_ref = function(arg0) {
-        const ret = getObject(arg0);
-        return addHeapObject(ret);
-    };
     imports.wbg.__wbindgen_debug_string = function(arg0, arg1) {
         const ret = debugString(getObject(arg1));
         const ptr1 = passStringToWasm0(ret, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
@@ -468,16 +683,24 @@ function __wbg_get_imports() {
     imports.wbg.__wbindgen_throw = function(arg0, arg1) {
         throw new Error(getStringFromWasm0(arg0, arg1));
     };
+    imports.wbg.__wbindgen_module = function() {
+        const ret = __wbg_init.__wbindgen_wasm_module;
+        return addHeapObject(ret);
+    };
     imports.wbg.__wbindgen_memory = function() {
         const ret = wasm.memory;
         return addHeapObject(ret);
     };
+    imports.wbg.__wbg_startWorkers_2ee336a9694dda13 = function(arg0, arg1, arg2) {
+        const ret = startWorkers(takeObject(arg0), takeObject(arg1), wbg_rayon_PoolBuilder.__wrap(arg2));
+        return addHeapObject(ret);
+    };
 
     return imports;
 }
 
 function __wbg_init_memory(imports, maybe_memory) {
-
+    imports.wbg.memory = maybe_memory || new WebAssembly.Memory({initial:18,maximum:16384,shared:true});
 }
 
 function __wbg_finalize_init(instance, module) {
@@ -487,16 +710,16 @@ function __wbg_finalize_init(instance, module) {
     cachedInt32Memory0 = null;
     cachedUint8Memory0 = null;
 
-
+    wasm.__wbindgen_start();
     return wasm;
 }
 
-function initSync(module) {
+function initSync(module, maybe_memory) {
     if (wasm !== undefined) return wasm;
 
     const imports = __wbg_get_imports();
 
-    __wbg_init_memory(imports);
+    __wbg_init_memory(imports, maybe_memory);
 
     if (!(module instanceof WebAssembly.Module)) {
         module = new WebAssembly.Module(module);
@@ -507,7 +730,7 @@ function initSync(module) {
     return __wbg_finalize_init(instance, module);
 }
 
-async function __wbg_init(input) {
+async function __wbg_init(input, maybe_memory) {
     if (wasm !== undefined) return wasm;
 
     if (typeof input === 'undefined') {
@@ -519,7 +742,7 @@ async function __wbg_init(input) {
         input = fetch(input);
     }
 
-    __wbg_init_memory(imports);
+    __wbg_init_memory(imports, maybe_memory);
 
     const { instance, module } = await __wbg_load(await input, imports);