umap.py

import umap
from sklearn.preprocessing import MinMaxScaler


def get_scaled_umap_embeddings(features, neighbour, distance):
    
    embedding = umap.UMAP(n_neighbors=neighbour,
                          min_dist=distance,
                          metric='correlation').fit_transform(features)
    scaler = MinMaxScaler()
    scaler.fit(embedding)
    return scaler.transform(embedding)


umap_embeddings_mfccs = []
umap_embeddings_wavenet = []
neighbours = [5, 10, 15, 30, 50]
distances = [0.000, 0.001, 0.01, 0.1, 0.5]

for neighbour in neighbours:
    for distance in distances:
        umap_mfccs = get_scaled_umap_embeddings(mfcc_features,
                                                neighbour,
                                                distance)
        umap_wavenet = get_scaled_umap_embeddings(wavenet_features,
                                                  neighbour,
                                                  distance)

Hey I get an error here: AttributeError: module 'umap' has no attribute 'UMAP'.. Any ideas?