本文整理汇总了Python中db.Database.make_train_test_matrix方法的典型用法代码示例。如果您正苦于以下问题：Python Database.make_train_test_matrix方法的具体用法？Python Database.make_train_test_matrix怎么用？Python Database.make_train_test_matrix使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类db.Database的用法示例。
在下文中一共展示了Database.make_train_test_matrix方法的2个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: sample_exp
# 需要导入模块: from db import Database [as 别名]
# 或者: from db.Database import make_train_test_matrix [as 别名]
def sample_exp():
# step 1: binary classification
libfm = LibFM()
db = Database('movielens', 718, 8928, 'ratings.csv', 'movies.attr')
db.load_data()
db.make_train_test_matrix()
db.add_negative_data()
db.dump_libfm_data('train_step1.libfm', 'test_step1.libfm')示例2: sample_exp
# 需要导入模块: from db import Database [as 别名]
# 或者: from db.Database import make_train_test_matrix [as 别名]
def sample_exp():
# making data from origin format to libfm format
#db = Database('movielens', 718, 8928, 'ratings.csv', 'movies.attr')
db = Database('amazon', 1000, 5000, 'cands_book.rating', 'cands_book.item')
db.load_data()
db.make_train_test_matrix()
libfm = LibFM()
# step 1: classification to predict interest
db.dump_libfm_data('train_step1.libfm', 'test_step1.libfm', add_negative=True, binary=True)
libfm.run('c', 'train_step1.libfm', 'test_step1.libfm', 'pred.libfm', iter_num=1)
step1_pred_list = db.load_pred_list('pred.libfm', 'c')
# TODO: merge accuracy accessing into libfm.py and call it here
# step 2: regression to predict possible rating
db.dump_libfm_data('train_step2.libfm', 'test_step2.libfm', add_negative=False, binary=False)
libfm.run('r', 'train_step1.libfm', 'test_step1.libfm', 'pred.libfm', iter_num=1)
step2_pred_list = db.load_pred_list('pred.libfm', 'r')