Home
last modified time | relevance | path

Searched refs:predictResult (Results 1 – 12 of 12) sorted by relevance

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/scala-package/infer/src/test/scala/org/apache/mxnet/infer/
H A DClassifierSuite.scala113 val predictResult : IndexedSeq[Array[Float]] = constant
124 assertResult(predictResult(0).sortBy(-_)) {
147 assertResult(predictResult(0).sortBy(-_)) {
168 assertResult(predictResult(0)) {
190 assertResult(predictResult(0)) {
199 val predictResult: IndexedSeq[Array[Float]] = constant
212 assert(predictResult.size == result.size)
214 for(i <- predictResult.indices) {
215 assertResult(predictResult(i)) {
241 assert(predictResult.size == result.size)
[all …]
H A DObjectDetectorSuite.scala91 val predictResult: IndexedSeq[IndexedSeq[(String, Array[Float])]] = constant
106 for (idx <- predictResult(0).indices) {
107 assert(predictResult(0)(idx)._1 == result(0)(idx)._1)
108 for (arridx <- predictResult(0)(idx)._2.indices) {
109 assert(predictResult(0)(idx)._2(arridx) == result(0)(idx)._2(arridx))
141 val predictResult: IndexedSeq[IndexedSeq[(String, Array[Float])]] = constant
160 for (preidx <- predictResult.indices) {
161 for (idx <- predictResult(preidx).indices) {
162 assert(predictResult(preidx)(idx)._1 == result(preidx)(idx)._1)
163 for (arridx <- predictResult(preidx)(idx)._2.indices) {
[all …]
H A DPredictorSuite.scala73 val predictResult = IndexedSeq(NDArray.ones(Shape(1, 3, 2, 2))) constant
77 Mockito.doReturn(predictResult).when(testPredictor.mockModule)
101 val predictResult = IndexedSeq(NDArray.ones(Shape(1, 3, 2, 2), dtype = DType.Float64)) constant
105 Mockito.doReturn(predictResult).when(testPredictor.mockModule)
130 val predictResult = IndexedSeq(NDArray.ones(Shape(1, 3, 2, 2))) constant
134 Mockito.doReturn(predictResult).when(testPredictor.mockModule)
H A DImageClassifierSuite.scala105 val predictResult: IndexedSeq[IndexedSeq[(String, Float)]] = constant
110 predictResult(i).map(_._2).toArray
/dports/misc/mxnet/incubator-mxnet-1.9.0/scala-package/infer/src/test/scala/org/apache/mxnet/infer/
H A DClassifierSuite.scala113 val predictResult : IndexedSeq[Array[Float]] = constant
124 assertResult(predictResult(0).sortBy(-_)) {
147 assertResult(predictResult(0).sortBy(-_)) {
168 assertResult(predictResult(0)) {
190 assertResult(predictResult(0)) {
199 val predictResult: IndexedSeq[Array[Float]] = constant
212 assert(predictResult.size == result.size)
214 for(i <- predictResult.indices) {
215 assertResult(predictResult(i)) {
241 assert(predictResult.size == result.size)
[all …]
H A DObjectDetectorSuite.scala91 val predictResult: IndexedSeq[IndexedSeq[(String, Array[Float])]] = constant
106 for (idx <- predictResult(0).indices) {
107 assert(predictResult(0)(idx)._1 == result(0)(idx)._1)
108 for (arridx <- predictResult(0)(idx)._2.indices) {
109 assert(predictResult(0)(idx)._2(arridx) == result(0)(idx)._2(arridx))
141 val predictResult: IndexedSeq[IndexedSeq[(String, Array[Float])]] = constant
160 for (preidx <- predictResult.indices) {
161 for (idx <- predictResult(preidx).indices) {
162 assert(predictResult(preidx)(idx)._1 == result(preidx)(idx)._1)
163 for (arridx <- predictResult(preidx)(idx)._2.indices) {
[all …]
H A DPredictorSuite.scala73 val predictResult = IndexedSeq(NDArray.ones(Shape(1, 3, 2, 2))) constant
77 Mockito.doReturn(predictResult).when(testPredictor.mockModule)
101 val predictResult = IndexedSeq(NDArray.ones(Shape(1, 3, 2, 2), dtype = DType.Float64)) constant
105 Mockito.doReturn(predictResult).when(testPredictor.mockModule)
130 val predictResult = IndexedSeq(NDArray.ones(Shape(1, 3, 2, 2))) constant
134 Mockito.doReturn(predictResult).when(testPredictor.mockModule)
H A DImageClassifierSuite.scala105 val predictResult: IndexedSeq[IndexedSeq[(String, Float)]] = constant
110 predictResult(i).map(_._2).toArray
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/scala-package/infer/src/main/scala/org/apache/mxnet/infer/
H A DObjectDetector.scala102 val predictResult = predictor.predictWithNDArray(input)(0).asInContext(Context.cpu()) constant
104 var batchResult = new ParArray[IndexedSeq[(String, Array[Float])]](predictResult.shape(0))
105 (0 until predictResult.shape(0)).toArray.par.foreach( i => {
106 val r = predictResult.at(i)
110 handler.execute(predictResult.dispose())
126 val predictResult = (0 until length).toArray.par.flatMap( i => { constant
140 var sortedIndices = predictResult.zipWithIndex.sortBy(-_._1(1)).map(_._2)
144 => (synset(predictResult(idx)(0).toInt),
145 predictResult(idx).takeRight(5))).toIndexedSeq
147 result = predictResult.map(ele
H A DClassifier.scala109 val predictResult = predictor.predict(input)(0) constant
114 val sortedIndex = predictResult.zipWithIndex.sortBy(-_._1).map(_._2).take(topK.get)
115 result = sortedIndex.map(i => (synset(i), predictResult(i))).toIndexedSeq
117 result = synset.zip(predictResult).toIndexedSeq
148 val predictResult = predictResultPar.toArray constant
153 val sortedIndices = predictResult.map(r =>
158 (synset(sIndx), predictResult(i)(sIndx))).toIndexedSeq
161 for (i <- predictResult.indices) {
162 result += synset.zip(predictResult(i)).toIndexedSeq
/dports/misc/mxnet/incubator-mxnet-1.9.0/scala-package/infer/src/main/scala/org/apache/mxnet/infer/
H A DObjectDetector.scala102 val predictResult = predictor.predictWithNDArray(input)(0).asInContext(Context.cpu()) constant
104 var batchResult = new ParArray[IndexedSeq[(String, Array[Float])]](predictResult.shape(0))
105 (0 until predictResult.shape(0)).toArray.par.foreach( i => {
106 val r = predictResult.at(i)
110 handler.execute(predictResult.dispose())
126 val predictResult = (0 until length).toArray.par.flatMap( i => { constant
140 var sortedIndices = predictResult.zipWithIndex.sortBy(-_._1(1)).map(_._2)
144 => (synset(predictResult(idx)(0).toInt),
145 predictResult(idx).takeRight(5))).toIndexedSeq
147 result = predictResult.map(ele
H A DClassifier.scala109 val predictResult = predictor.predict(input)(0) constant
114 val sortedIndex = predictResult.zipWithIndex.sortBy(-_._1).map(_._2).take(topK.get)
115 result = sortedIndex.map(i => (synset(i), predictResult(i))).toIndexedSeq
117 result = synset.zip(predictResult).toIndexedSeq
148 val predictResult = predictResultPar.toArray constant
153 val sortedIndices = predictResult.map(r =>
158 (synset(sIndx), predictResult(i)(sIndx))).toIndexedSeq
161 for (i <- predictResult.indices) {
162 result += synset.zip(predictResult(i)).toIndexedSeq