Package | Description |
---|---|
org.apache.ignite.ml.dataset |
Base package for machine learning dataset classes.
|
org.apache.ignite.ml.dataset.impl.bootstrapping |
Base package for bootstrapped implementation of machine learning dataset.
|
org.apache.ignite.ml.dataset.impl.cache |
Base package for cache based implementation of machine learning dataset.
|
org.apache.ignite.ml.dataset.impl.cache.util |
Contains util classes used in cache based implementation of dataset.
|
org.apache.ignite.ml.dataset.impl.local |
Base package for local implementation of machine learning dataset.
|
org.apache.ignite.ml.dataset.primitive |
Package that contains basic primitives build on top of
Dataset . |
org.apache.ignite.ml.dataset.primitive.builder.data |
Contains partition
data builders. |
org.apache.ignite.ml.knn |
Contains main APIs for kNN algorithms.
|
org.apache.ignite.ml.math.isolve.lsqr |
Contains LSQR algorithm implementation.
|
org.apache.ignite.ml.recommendation.util |
Contains util classes used in recommendation system framework.
|
org.apache.ignite.ml.structures.partition |
Contains internal APIs for dataset partitioned labeled datasets.
|
org.apache.ignite.ml.tree.data |
Contains data and data builder required for decision tree trainers built on top of partition based dataset.
|
Modifier and Type | Method and Description |
---|---|
default <D2 extends AutoCloseable> |
PartitionDataBuilder.andThen(IgniteBiFunction<D,C,D2> fun)
Makes a composed partition
data builder that first builds a data and then applies the specified
function on the result. |
Modifier and Type | Method and Description |
---|---|
<C extends Serializable,D extends AutoCloseable> |
DatasetBuilder.build(LearningEnvironmentBuilder envBuilder,
PartitionContextBuilder<K,V,C> partCtxBuilder,
PartitionDataBuilder<K,V,C,D> partDataBuilder,
LearningEnvironment localLearningEnv)
Constructs a new instance of
Dataset that includes allocation required data structures and
initialization of context part of partitions. |
static <K,V,C extends Serializable,D extends AutoCloseable> |
DatasetFactory.create(DatasetBuilder<K,V> datasetBuilder,
LearningEnvironmentBuilder envBuilder,
PartitionContextBuilder<K,V,C> partCtxBuilder,
PartitionDataBuilder<K,V,C,D> partDataBuilder,
LearningEnvironment environment)
Creates a new instance of distributed dataset using the specified
partCtxBuilder and partDataBuilder . |
static <K,V,C extends Serializable,D extends AutoCloseable> |
DatasetFactory.create(DatasetBuilder<K,V> datasetBuilder,
PartitionContextBuilder<K,V,C> partCtxBuilder,
PartitionDataBuilder<K,V,C,D> partDataBuilder)
Creates a new instance of distributed dataset using the specified
partCtxBuilder and partDataBuilder . |
static <K,V,C extends Serializable,D extends AutoCloseable> |
DatasetFactory.create(Ignite ignite,
IgniteCache<K,V> upstreamCache,
LearningEnvironmentBuilder envBuilder,
PartitionContextBuilder<K,V,C> partCtxBuilder,
PartitionDataBuilder<K,V,C,D> partDataBuilder,
LearningEnvironment environment)
Creates a new instance of distributed dataset using the specified
partCtxBuilder and partDataBuilder . |
static <K,V,C extends Serializable,D extends AutoCloseable> |
DatasetFactory.create(Ignite ignite,
IgniteCache<K,V> upstreamCache,
PartitionContextBuilder<K,V,C> partCtxBuilder,
PartitionDataBuilder<K,V,C,D> partDataBuilder)
Creates a new instance of distributed dataset using the specified
partCtxBuilder and partDataBuilder . |
static <K,V,C extends Serializable,D extends AutoCloseable> |
DatasetFactory.create(Map<K,V> upstreamMap,
LearningEnvironmentBuilder envBuilder,
int partitions,
PartitionContextBuilder<K,V,C> partCtxBuilder,
PartitionDataBuilder<K,V,C,D> partDataBuilder,
LearningEnvironment environment)
Creates a new instance of local dataset using the specified
partCtxBuilder and partDataBuilder . |
Modifier and Type | Class and Description |
---|---|
class |
BootstrappedDatasetBuilder<K,V>
Builder for bootstrapped dataset.
|
Modifier and Type | Method and Description |
---|---|
<C extends Serializable,D extends AutoCloseable> |
CacheBasedDatasetBuilder.build(LearningEnvironmentBuilder envBuilder,
PartitionContextBuilder<K,V,C> partCtxBuilder,
PartitionDataBuilder<K,V,C,D> partDataBuilder,
LearningEnvironment localLearningEnv)
Constructs a new instance of
Dataset that includes allocation required data structures and
initialization of context part of partitions. |
Constructor and Description |
---|
CacheBasedDataset(Ignite ignite,
IgniteCache<K,V> upstreamCache,
IgniteBiPredicate<K,V> filter,
UpstreamTransformerBuilder upstreamTransformerBuilder,
IgniteCache<Integer,C> datasetCache,
LearningEnvironmentBuilder envBuilder,
PartitionDataBuilder<K,V,C,D> partDataBuilder,
UUID datasetId,
boolean upstreamKeepBinary,
LearningEnvironment localLearningEnv,
int retriesCnt)
Constructs a new instance of dataset based on Ignite Cache, which is used as
upstream and as reliable storage for
partition context as well. |
Modifier and Type | Method and Description |
---|---|
static <K,V,C extends Serializable,D extends AutoCloseable> |
ComputeUtils.getData(Ignite ignite,
String upstreamCacheName,
IgniteBiPredicate<K,V> filter,
UpstreamTransformerBuilder transformerBuilder,
String datasetCacheName,
UUID datasetId,
PartitionDataBuilder<K,V,C,D> partDataBuilder,
LearningEnvironment env,
boolean isKeepBinary)
Extracts partition
data from the local storage, if it's not found in local storage recovers this data from a partition upstream and context . |
Modifier and Type | Method and Description |
---|---|
<C extends Serializable,D extends AutoCloseable> |
LocalDatasetBuilder.build(LearningEnvironmentBuilder envBuilder,
PartitionContextBuilder<K,V,C> partCtxBuilder,
PartitionDataBuilder<K,V,C,D> partDataBuilder,
LearningEnvironment learningEnvironment)
Constructs a new instance of
Dataset that includes allocation required data structures and
initialization of context part of partitions. |
Modifier and Type | Class and Description |
---|---|
class |
FeatureMatrixWithLabelsOnHeapDataBuilder<K,V,C extends Serializable,CO extends Serializable>
A partition
data builder that makes DecisionTreeData . |
Modifier and Type | Class and Description |
---|---|
class |
SimpleDatasetDataBuilder<K,V,C extends Serializable,CO extends Serializable>
A partition
data builder that makes SimpleDatasetData . |
class |
SimpleLabeledDatasetDataBuilder<K,V,C extends Serializable>
A partition
data builder that makes SimpleLabeledDatasetData . |
Modifier and Type | Class and Description |
---|---|
class |
KNNPartitionDataBuilder<K,V>
Partition data builder for KNN algorithms based on
SpatialIndex . |
Constructor and Description |
---|
LSQROnHeap(DatasetBuilder<K,V> datasetBuilder,
LearningEnvironmentBuilder envBuilder,
PartitionDataBuilder<K,V,LSQRPartitionContext,SimpleLabeledDatasetData> partDataBuilder,
LearningEnvironment localLearningEnv)
Constructs a new instance of OnHeap LSQR algorithm implementation.
|
Modifier and Type | Class and Description |
---|---|
class |
RecommendationBinaryDatasetDataBuilder
Recommendation binary dataset data builder.
|
class |
RecommendationDatasetDataBuilder<K,O extends Serializable,S extends Serializable,Z extends ObjectSubjectRatingTriplet<O,S>>
A partition
data builder that makes SimpleDatasetData . |
Modifier and Type | Class and Description |
---|---|
class |
LabeledDatasetPartitionDataBuilderOnHeap<K,V,C extends Serializable>
Partition data builder that builds
LabeledVectorSet . |
class |
LabelPartitionDataBuilderOnHeap<K,V,C extends Serializable>
Partition data builder that builds
LabelPartitionDataOnHeap . |
Modifier and Type | Class and Description |
---|---|
class |
DecisionTreeDataBuilder<K,V,C extends Serializable>
A partition
data builder that makes DecisionTreeData . |
GridGain In-Memory Computing Platform : ver. 8.9.14 Release Date : November 5 2024