| Package | Description |
|---|---|
| org.apache.ignite.ml.dataset |
Base package for machine learning dataset classes.
|
| org.apache.ignite.ml.dataset.primitive.builder.context |
Contains partition
context builders. |
| org.apache.ignite.ml.knn |
Contains main APIs for kNN algorithms.
|
| org.apache.ignite.ml.knn.classification |
Contains main APIs for kNN classification algorithms.
|
| org.apache.ignite.ml.knn.regression |
Contains helper classes for kNN regression algorithms.
|
| org.apache.ignite.ml.nn |
Contains neural networks and related classes.
|
| org.apache.ignite.ml.tree.leaf |
Root package for decision trees leaf builders.
|
| Modifier and Type | Method and Description |
|---|---|
static <K,V> SimpleDataset<EmptyContext> |
DatasetFactory.createSimpleDataset(DatasetBuilder<K,V> datasetBuilder,
IgniteBiFunction<K,V,double[]> featureExtractor)
Creates a new instance of distributed
SimpleDataset using the specified featureExtractor. |
static <K,V> SimpleDataset<EmptyContext> |
DatasetFactory.createSimpleDataset(Ignite ignite,
IgniteCache<K,V> upstreamCache,
IgniteBiFunction<K,V,double[]> featureExtractor)
Creates a new instance of distributed
SimpleDataset using the specified featureExtractor. |
static <K,V> SimpleDataset<EmptyContext> |
DatasetFactory.createSimpleDataset(Map<K,V> upstreamMap,
int partitions,
IgniteBiFunction<K,V,double[]> featureExtractor)
Creates a new instance of local
SimpleDataset using the specified featureExtractor. |
static <K,V> SimpleLabeledDataset<EmptyContext> |
DatasetFactory.createSimpleLabeledDataset(DatasetBuilder<K,V> datasetBuilder,
IgniteBiFunction<K,V,double[]> featureExtractor,
IgniteBiFunction<K,V,double[]> lbExtractor)
Creates a new instance of distributed
SimpleLabeledDataset using the specified featureExtractor
and lbExtractor. |
static <K,V> SimpleLabeledDataset<EmptyContext> |
DatasetFactory.createSimpleLabeledDataset(Ignite ignite,
IgniteCache<K,V> upstreamCache,
IgniteBiFunction<K,V,double[]> featureExtractor,
IgniteBiFunction<K,V,double[]> lbExtractor)
Creates a new instance of distributed
SimpleLabeledDataset using the specified featureExtractor
and lbExtractor. |
static <K,V> SimpleLabeledDataset<EmptyContext> |
DatasetFactory.createSimpleLabeledDataset(Map<K,V> upstreamMap,
int partitions,
IgniteBiFunction<K,V,double[]> featureExtractor,
IgniteBiFunction<K,V,double[]> lbExtractor)
Creates a new instance of local
SimpleLabeledDataset using the specified featureExtractor
and lbExtractor. |
| Modifier and Type | Method and Description |
|---|---|
EmptyContext |
EmptyContextBuilder.build(Iterator<UpstreamEntry<K,V>> upstreamData,
long upstreamDataSize)
Builds a new partition
context from an upstream data. |
| Modifier and Type | Method and Description |
|---|---|
static <K,V> Dataset<EmptyContext,LabeledDataset<Double,LabeledVector>> |
KNNUtils.buildDataset(DatasetBuilder<K,V> datasetBuilder,
IgniteBiFunction<K,V,double[]> featureExtractor,
IgniteBiFunction<K,V,Double> lbExtractor)
Builds dataset.
|
| Constructor and Description |
|---|
KNNClassificationModel(Dataset<EmptyContext,LabeledDataset<Double,LabeledVector>> dataset)
Builds the model via prepared dataset.
|
| Constructor and Description |
|---|
KNNRegressionModel(Dataset<EmptyContext,LabeledDataset<Double,LabeledVector>> dataset)
Builds the model via prepared dataset.
|
| Constructor and Description |
|---|
MLPTrainer(IgniteFunction<Dataset<EmptyContext,SimpleLabeledDatasetData>,MLPArchitecture> archSupplier,
IgniteFunction<Vector,IgniteDifferentiableVectorToDoubleFunction> loss,
UpdatesStrategy<? super MultilayerPerceptron,P> updatesStgy,
int maxIterations,
int batchSize,
int locIterations,
long seed)
Constructs a new instance of multilayer perceptron trainer.
|
| Modifier and Type | Method and Description |
|---|---|
DecisionTreeLeafNode |
DecisionTreeLeafBuilder.createLeafNode(Dataset<EmptyContext,DecisionTreeData> dataset,
TreeFilter pred)
Creates new leaf node for given dataset and node predicate.
|
DecisionTreeLeafNode |
MostCommonDecisionTreeLeafBuilder.createLeafNode(Dataset<EmptyContext,DecisionTreeData> dataset,
TreeFilter pred)
Creates new leaf node for given dataset and node predicate.
|
DecisionTreeLeafNode |
MeanDecisionTreeLeafBuilder.createLeafNode(Dataset<EmptyContext,DecisionTreeData> dataset,
TreeFilter pred)
Creates new leaf node for given dataset and node predicate.
|
Follow @ApacheIgnite
Ignite Fabric : ver. 2.5.0 Release Date : May 23 2018