Scikit-learn based interface with fit/predict functions added for easier usage. Added labels_ and centroids_ attributes of two classes KShapeClustering CPU and KShapeClustering GPU.
Added n_jobs parameter which allows to adjust number of processes in multiprocessing calculations
Fix issue #14 which used depreciated symeig function. It was changed to torch.linalg.eigh which is equivalent.
Fix issue with nan_to_num_cuda error for complex numbers that were returned by eigh function
Fix issue with multiprocessing pool added in _extract_shape function. The multiprocessing at this point was removed which speeds up computations a lot
Following changes were made to the original code:
labels_andcentroids_attributes of two classesKShapeClustering CPUandKShapeClustering GPU.n_jobsparameter which allows to adjust number of processes in multiprocessing calculationssymeigfunction. It was changed totorch.linalg.eighwhich is equivalent.nan_to_num_cudaerror for complex numbers that were returned byeighfunction_extract_shapefunction. The multiprocessing at this point was removed which speeds up computations a lot