org.apache.spark.mllib.linalg.distributed

Class GridPartitioner

Object

org.apache.spark.Partitioner

org.apache.spark.mllib.linalg.distributed.GridPartitioner

All Implemented Interfaces:

java.io.Serializable

public class GridPartitioner
extends Partitioner

A grid partitioner, which uses a regular grid to partition coordinates.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
GridPartitioner(int rows, int cols, int rowsPerPart, int colsPerPart)

Method Summary

Methods

Modifier and Type	Method and Description
static GridPartitioner	apply(int rows, int cols, int suggestedNumPartitions) Creates a new GridPartitioner instance with the input suggested number of partitions.
static GridPartitioner	apply(int rows, int cols, int rowsPerPart, int colsPerPart) Creates a new GridPartitioner instance.
int	cols()
int	colsPerPart()
boolean	equals(Object obj)
int	getPartition(Object key) Returns the index of the partition the input coordinate belongs to.
int	numPartitions()
int	rows()
int	rowsPerPart()

Methods inherited from class org.apache.spark.Partitioner

defaultPartitioner

Methods inherited from class Object

getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

GridPartitioner

public GridPartitioner(int rows,
               int cols,
               int rowsPerPart,
               int colsPerPart)

Method Detail

apply

public static GridPartitioner apply(int rows,
                    int cols,
                    int rowsPerPart,
                    int colsPerPart)

Creates a new GridPartitioner instance.

apply

public static GridPartitioner apply(int rows,
                    int cols,
                    int suggestedNumPartitions)

Creates a new GridPartitioner instance with the input suggested number of partitions.

rows

public int rows()

cols

public int cols()

rowsPerPart

public int rowsPerPart()

colsPerPart

public int colsPerPart()

numPartitions

public int numPartitions()

Specified by:

numPartitions in class Partitioner

getPartition

public int getPartition(Object key)

Returns the index of the partition the input coordinate belongs to.

Specified by:

getPartition in class Partitioner

Parameters:

key - The coordinate (i, j) or a tuple (i, j, k), where k is the inner index used in multiplication. k is ignored in computing partitions.

Returns:

The index of the partition, which the coordinate belongs to.

equals

public boolean equals(Object obj)

Overrides:

equals in class Object