I get a model result with all Wi = NaN about fregata

[takun2s]

No, there is no limitation of the data size .
You could try to normalize the data .

[qf6101]

I encountered a similar overflow problem with SoftMax. And I fix the code as follows.

val margins = ws.map { w =>
  VectorUtil.wxpb(w,x,1d)
}
val maxMargin = margins.max
val adjustedMargins = margins.map { margin =>
  margin - maxMargin
}
val mes = adjustedMargins.map {...