NNSOM.som_gpu.SOMGpu.train

SOMGpu.train(x, init_neighborhood=3, epochs=200, steps=100, norm_func=None)[source]

Trains the SOM using the batch SOM algorithm on the input data x.

Parameters:

  • x (np.ndarray) – The input data to train the SOM with.

  • init_neighborhood (int, optional) – The initial neighborhood size.

  • epochs (int, optional) – The number of epochs to train for.

  • steps (int, optional) – The number of steps for training.

Return type:

None