org.deeplearning4j.nn.multilayer.MultiLayerNetwork.predict()方法的使用及代码示例

x33g5p2x  于2022-01-25 转载在 其他  
字(1.4k)|赞(0)|评价(0)|浏览(110)

本文整理了Java中org.deeplearning4j.nn.multilayer.MultiLayerNetwork.predict()方法的一些代码示例,展示了MultiLayerNetwork.predict()的具体用法。这些代码示例主要来源于Github/Stackoverflow/Maven等平台,是从一些精选项目中提取出来的代码,具有较强的参考意义,能在一定程度帮忙到你。MultiLayerNetwork.predict()方法的具体详情如下:
包路径:org.deeplearning4j.nn.multilayer.MultiLayerNetwork
类名称:MultiLayerNetwork
方法名:predict

MultiLayerNetwork.predict介绍

[英]Returns the predictions for each example in the dataset
[中]返回数据集中每个示例的预测

代码示例

代码示例来源:origin: deeplearning4j/dl4j-examples

List<String> allClassLabels = validateIterator.getLabels();
int labelIndex = validateDataSet.getLabels().argMax(1).getInt(0);
int[] predictedClasses = network.predict(validateDataSet.getFeatures());
String expectedResult = allClassLabels.get(labelIndex);
String modelPrediction = allClassLabels.get(predictedClasses[0]);

代码示例来源:origin: org.deeplearning4j/deeplearning4j-nn

/**
 * Return predicted label names
 *
 * @param dataSet to predict
 * @return the predicted labels for the dataSet
 */
@Override
public List<String> predict(org.nd4j.linalg.dataset.api.DataSet dataSet) {
  int[] intRet = predict(dataSet.getFeatures());
  List<String> ret = new ArrayList<>();
  for (int i = 0; i < intRet.length; i++) {
    ret.add(i, dataSet.getLabelName(intRet[i]));
  }
  return ret;
}

代码示例来源:origin: sjsdfg/dl4j-tutorials

normalizer.transform(indArray);
int[] values = network.predict(indArray);
String label = map.get(values[0]);

代码示例来源:origin: sjsdfg/dl4j-tutorials

int[] results = network.predict(imgNdarray);
System.out.println(Arrays.toString(results));

相关文章