座標マトリクスの列方向の最大値

Question

私はspark.mllib.linalg.distributed.CoordinateMatrixというマットを持っています。このスパース行列の行方向の最大値を計算したいと思います。私はmat.entriesがマットの内容をMatrixEntry(i,j,v)として保存していることを知っています。今私はマットの行の賢明な最大値を見つけることを望む。つまり、すべてのiに対して、すべてのjに対して最大値vを見つける必要があります。これを行うにはどうしたらいいですか？

これまでのところ、私はこれがあります。

val mat_RowMatrix = mat.toRowMatrix() // found function toRowMatrix on github 
val max_entries = mat_RowMatrix.rows.map{_.toArray.max} 

をしかし、私はARGMAX（最大のインデックス）と値ではなく自分自身を見つける必要があります。これについてのドキュメントはどこにありますか？私はmaxByを使わなければならないと思うが、それを使う方法は知らない。また、これを行うにはよりよい方法がありますか？

いくつかの方向性が役立ちます。

Answer 1

RowMatrix.rowsはRDD[Vector]となり、Vectorは既にargmax機能でこの機能を提供しています。だからあなたはすることができます：

val maxEntries = matRowMatrix.rows.map{_.argmax}