本文整理汇总了Java中org.nd4j.linalg.ops.transforms.Transforms.allCosineDistances方法的典型用法代码示例。如果您正苦于以下问题:Java Transforms.allCosineDistances方法的具体用法?Java Transforms.allCosineDistances怎么用?Java Transforms.allCosineDistances使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类org.nd4j.linalg.ops.transforms.Transforms
的用法示例。
在下文中一共展示了Transforms.allCosineDistances方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Java代码示例。
示例1: testAllDistances5_Large_Columns
import org.nd4j.linalg.ops.transforms.Transforms; //导入方法依赖的package包/类
@Test
public void testAllDistances5_Large_Columns() throws Exception {
INDArray initialX = Nd4j.create(2000, 5);
INDArray initialY = Nd4j.create(2000, 7);
for (int i = 0; i < initialX.columns(); i++) {
initialX.getColumn(i).assign(i + 1);
}
for (int i = 0; i < initialY.columns(); i++) {
initialY.getColumn(i).assign(i + 101);
}
INDArray result = Transforms.allCosineDistances(initialX, initialY, 0);
assertEquals(5 * 7, result.length());
for (int x = 0; x < initialX.columns(); x++) {
INDArray colX = initialX.getColumn(x).dup();
for (int y = 0; y < initialY.columns(); y++) {
double res = result.getDouble(x, y);
double exp = Transforms.cosineDistance(colX, initialY.getColumn(y).dup());
assertEquals("Failed for [" + x + ", " + y + "]", exp, res, 0.001);
}
}
}