标签： python machine-learning scikit-learn
我正在尝试在符合scikit-learn约定的模型类中包装一个名为“ GEMSEC: Embedding with Self Clustering”的新方法。
我了解了预测函数here，看来预测函数必须返回[n_samples,]或[n_samples, n_outputs]的数组。
[n_samples,]
[n_samples, n_outputs]
我正在实现的模型做了两件事：学习嵌入（表示）和聚类，并且我不知道从scikit-定义的predict函数中，预测函数可以返回什么？学习。
predict
预先感谢