reference.rst

API Reference

This is the class and function reference of Mars learn.

mars.learn

Datasets

mars.learn.datasets

Samples generator

mars.learn

datasets.make_blobs datasets.make_classification datasets.make_low_rank_matrix

Matrix Decomposition

mars.learn.decomposition

mars.learn

decomposition.PCA decomposition.TruncatedSVD

Metrics

mars.learn.metrics

mars.learn

Classification metrics

metrics.accuracy_score metrics.auc metrics.roc_curve

Pairwise metrics

mars.learn.metrics.pairwise

mars.learn

metrics.pairwise.cosine_similarity metrics.pairwise.cosine_distances metrics.pairwise.euclidean_distances metrics.pairwise.haversine_distances metrics.pairwise.manhattan_distances metrics.pairwise.rbf_kernel metrics.pairwise_distances

Splitter Functions

mars.learn

model_selection.train_test_split

Nearest Neighbors

mars.learn.neighbors

mars.learn

neighbors.NearestNeighbors

Preprocessing and Normalization

mars.learn.preprocessing

mars.learn

preprocessing.normalize

Semi-Supervised Learning

mars.learn.semi_supervised

mars.learn

semi_supervised.LabelPropagation

Utilities

mars.learn.utils

mars.learn

utils.assert_all_finite utils.check_X_y utils.check_array utils.check_consistent_length utils.multiclass.type_of_target utils.multiclass.is_multilabel utils.shuffle utils.validation.check_is_fitted utils.validation.column_or_1d

TensorFlow Integration

mars.learn.contrib.tensorflow

mars.learn

contrib.tensorflow.run_tensorflow_script

PyTorch Integration

mars.learn.contrib.tensorflow

mars.learn

contrib.pytorch.run_pytorch_script contrib.pytorch.MarsDataset contrib.pytorch.MarsDistributedSampler contrib.pytorch.MarsRandomSampler

XGBoost Integration

mars.learn.contrib.xgboost

mars.learn

contrib.xgboost.MarsDMatrix contrib.xgboost.train contrib.xgboost.predict contrib.xgboost.XGBClassifier contrib.xgboost.XGBRegressor

LightGBM Integration

mars.learn.contrib.lightgbm

mars.learn

contrib.lightgbm.LGBMClassifier contrib.lightgbm.LGBMRegressor contrib.lightgbm.LGBMRanker