本文整理汇总了Python中db.Database.dump_libfm_data方法的典型用法代码示例。如果您正苦于以下问题:Python Database.dump_libfm_data方法的具体用法?Python Database.dump_libfm_data怎么用?Python Database.dump_libfm_data使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类db.Database
的用法示例。
在下文中一共展示了Database.dump_libfm_data方法的2个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: sample_exp
# 需要导入模块: from db import Database [as 别名]
# 或者: from db.Database import dump_libfm_data [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 dump_libfm_data [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')