API References

Training API

RelationBlock	The RelationBlock Class.
MyFMRegressor	alias of myfm.gibbs.MyFMGibbsRegressor
MyFMClassifier	alias of myfm.gibbs.MyFMGibbsClassifier
MyFMGibbsRegressor(rank[, init_stdev, ...])
MyFMGibbsClassifier(rank[, init_stdev, ...])	Bayesian Factorization Machines for binary classification tasks.
MyFMOrderedProbit(rank[, init_stdev, ...])	Bayesian Factorization Machines for Ordinal Regression Tasks.
VariationalFMRegressor(rank[, init_stdev, ...])	Variational Inference for Regression Task.
VariationalFMClassifier(rank[, init_stdev, ...])	Variational Inference for Classification Task.

Benchmark Dataset

utils.benchmark_data.MovieLens100kDataManager([...])	The Data manager for MovieLens 100k dataset.
utils.benchmark_data.MovieLens1MDataManager([...])
utils.benchmark_data.MovieLens10MDataManager([...])

Utilities for Sparse Matrix Construction

utils.encoders.DataFrameEncoder()	Encode pandas.DataFrame into concatenated sparse matrices.
utils.encoders.CategoryValueToSparseEncoder(...)	The class to one-hot encode a List of items into a sparse matrix representation.
utils.encoders.MultipleValuesToSparseEncoder(...)	The class to N-hot encode a List of items into a sparse matrix representation.
utils.encoders.BinningEncoder(x[, n_percentiles])	The class to one-hot encode a List of numerical values into a sparse matrix representation by binning.