libmf 0.1.3 → 0.2.0

data/vendor/libmf/mf.def

@@ -1,21 +0,0 @@
-LIBRARY mf
-EXPORTS
-    mf_get_default_param    @1
-    read_problem    @2
-    mf_save_model    @3
-    mf_load_model    @4
-    mf_destroy_model    @5
-    mf_train    @6
-    mf_train_on_disk    @7
-    mf_train_with_validation    @8
-    mf_train_with_validation_on_disk    @9
-    mf_cross_validation    @10
-    mf_cross_validation_on_disk    @11
-    mf_predict    @12
-    calc_rmse    @13
-    calc_mae    @14
-    calc_gkl    @15
-    calc_logloss    @16
-    calc_accuracy    @17
-    calc_mpr    @18
-    calc_auc    @19

data/vendor/libmf/mf.h

@@ -1,130 +0,0 @@
-#ifndef _LIBMF_H
-#define _LIBMF_H
-
-#include <string>
-#include <utility>
-
-#ifdef __cplusplus
-extern "C"
-{
-
-namespace mf
-{
-#endif
-
-typedef float mf_float;
-typedef double mf_double;
-typedef int mf_int;
-typedef long long mf_long;
-
-enum {P_L2_MFR=0, P_L1_MFR=1, P_KL_MFR=2, P_LR_MFC=5, P_L2_MFC=6, P_L1_MFC=7,
-      P_ROW_BPR_MFOC=10, P_COL_BPR_MFOC=11, P_L2_MFOC=12};
-enum {RMSE=0, MAE=1, GKL=2, LOGLOSS=5, ACC=6, ROW_MPR=10, COL_MPR=11,
-      ROW_AUC=12, COL_AUC=13};
-
-struct mf_node
-{
-    mf_int u;
-    mf_int v;
-    mf_float r;
-};
-
-struct mf_problem
-{
-    mf_int m;
-    mf_int n;
-    mf_long nnz;
-    struct mf_node *R;
-};
-
-struct mf_parameter
-{
-    mf_int fun;
-    mf_int k;
-    mf_int nr_threads;
-    mf_int nr_bins;
-    mf_int nr_iters;
-    mf_float lambda_p1;
-    mf_float lambda_p2;
-    mf_float lambda_q1;
-    mf_float lambda_q2;
-    mf_float eta;
-    mf_float alpha;
-    mf_float c;
-    bool do_nmf;
-    bool quiet;
-    bool copy_data;
-};
-
-struct mf_parameter mf_get_default_param();
-
-struct mf_model
-{
-    mf_int fun;
-    mf_int m;
-    mf_int n;
-    mf_int k;
-    mf_float b;
-    mf_float *P;
-    mf_float *Q;
-};
-
-mf_problem read_problem(std::string path);
-
-mf_int mf_save_model(struct mf_model const *model, char const *path);
-
-struct mf_model* mf_load_model(char const *path);
-
-void mf_destroy_model(struct mf_model **model);
-
-struct mf_model* mf_train(
-    struct mf_problem const *prob,
-    struct mf_parameter param);
-
-struct mf_model* mf_train_on_disk(
-    char const *tr_path,
-    struct mf_parameter param);
-
-struct mf_model* mf_train_with_validation(
-    struct mf_problem const *tr,
-    struct mf_problem const *va,
-    struct mf_parameter param);
-
-struct mf_model* mf_train_with_validation_on_disk(
-    char const *tr_path,
-    char const *va_path,
-    struct mf_parameter param);
-
-mf_double mf_cross_validation(
-    struct mf_problem const *prob,
-    mf_int nr_folds,
-    struct mf_parameter param);
-
-mf_double mf_cross_validation_on_disk(
-    char const *prob,
-    mf_int nr_folds,
-    mf_parameter param);
-
-mf_float mf_predict(struct mf_model const *model, mf_int u, mf_int v);
-
-mf_double calc_rmse(mf_problem *prob, mf_model *model);
-
-mf_double calc_mae(mf_problem *prob, mf_model *model);
-
-mf_double calc_gkl(mf_problem *prob, mf_model *model);
-
-mf_double calc_logloss(mf_problem *prob, mf_model *model);
-
-mf_double calc_accuracy(mf_problem *prob, mf_model *model);
-
-mf_double calc_mpr(mf_problem *prob, mf_model *model, bool transpose);
-
-mf_double calc_auc(mf_problem *prob, mf_model *model, bool transpose);
-
-#ifdef __cplusplus
-} // namespace mf
-
-} // extern "C"
-#endif
-
-#endif // _LIBMF_H