本文整理汇总了Java中org.datavec.api.records.reader.RecordReader.setConf方法的典型用法代码示例。如果您正苦于以下问题:Java RecordReader.setConf方法的具体用法?Java RecordReader.setConf怎么用?Java RecordReader.setConf使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类org.datavec.api.records.reader.RecordReader
的用法示例。
在下文中一共展示了RecordReader.setConf方法的2个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Java代码示例。
示例1: testMiniBatches
import org.datavec.api.records.reader.RecordReader; //导入方法依赖的package包/类
@Test
public void testMiniBatches() throws Exception {
log.info("Setting up Spark Context...");
JavaRDD<String> lines = sc.textFile(new ClassPathResource("svmLight/iris_svmLight_0.txt")
.getTempFileFromArchive().toURI().toString()).cache();
long count = lines.count();
assertEquals(300, count);
// gotta map this to a Matrix/INDArray
RecordReader rr = new SVMLightRecordReader();
Configuration c = new Configuration();
c.set(SVMLightRecordReader.NUM_FEATURES, "5");
rr.setConf(c);
JavaRDD<DataSet> points = lines.map(new RecordReaderFunction(rr, 4, 3)).cache();
count = points.count();
assertEquals(300, count);
JavaRDD<DataSet> miniBatches = new RDDMiniBatches(10, points).miniBatchesJava();
count = miniBatches.count();
assertEquals(30, count);
lines.unpersist();
points.unpersist();
miniBatches.map(new DataSetAssertionFunction());
}
示例2: setConf
import org.datavec.api.records.reader.RecordReader; //导入方法依赖的package包/类
@Override
public void setConf(Configuration conf) {
for (RecordReader reader : readers) {
reader.setConf(conf);
}
}