标签： machine-learning gpu google-colaboratory recommendation-engine cosine-similarity
因此，我有一个巨大的tfidf矩阵，其中有超过一百万条记录，我想找到此矩阵与其自身的余弦相似度。我正在使用colab来运行代码，但是我不确定如何最好地利用colab提供的gpu。
依次运行代码-
tfidf_matrix = tf.fit_transform(df['categories'])
cosine_similarities = linear_kernel(matrix, matrix)
有没有办法使用jit或任何其他方式并行化代码？