data['probabilities'] = best_lda_model\
                        .transform(sessions_vectorized).tolist()

def get_cluster(probabilities):
    x = np.argmax(probabilities)
    return x

data['cluster'] = data['probabilities'].apply(get_cluster)