'rmse': rmse, 'lrate': current_l_w } config_result['results'].append(iteration_result) print(iteration_str.format(j, k, current_l_w, momentum, mae, rmse)) with open('experiments/{}_{}.json'.format(config_name, name), 'wt') as res_output: res_output.write(json.dumps(config_result, indent=4)) W,V,H = rbm.get_weights() print H if __name__ == "__main__": experiments = read_experiment(sys.argv[1]) for experiment in experiments: name = experiment['name'] train_path = experiment['train_path'] test_path = experiment['test_path'] sep = experiment['sep'] configs = experiment['configs'] all_users, all_movies, tests = load_dataset(train_path, test_path, sep, user_based=True) for config in configs: run(name, train_path, config, all_users, all_movies, tests, None, sep)
} config_result['results'].append(iteration_result) print(iteration_str.format(j, k, current_l_w, momentum, mae, rmse)) with open('{}_{}.json'.format(config_name, name), 'wt') as res_output: res_output.write(json.dumps(config_result, indent=4)) w = rbm.weights.eval() np.save('weights', w) if __name__ == "__main__": path = "../ubased.json" experiments = read_experiment(path) for experiment in experiments: name = experiment['name'] train_path = experiment['train_path'] test_path = experiment['test_path'] sep = experiment['sep'] configs = experiment['configs'] all_users, all_movies, tests = load_dataset(train_path, test_path, sep, user_based=True) # print all_users # print len(all_users)
'k': k, 'momentum': momentum, 'mae': mae, 'rmse': rmse, 'lrate': current_l_w } config_result['results'].append(iteration_result) print(iteration_str.format(j, k, current_l_w, momentum, mae, rmse)) with open('{}_{}.json'.format(config_name, name), 'wt') as res_output: res_output.write(json.dumps(config_result, indent=4)) if __name__ == "__main__": experiments = read_experiment(sys.argv[1]) for experiment in experiments: name = experiment['name'] train_path = experiment['train_path'] test_path = experiment['test_path'] sep = experiment['sep'] configs = experiment['configs'] all_users, all_movies, tests = load_dataset(train_path, test_path, sep, user_based=True) for config in configs: run(name, train_path, config, all_users, all_movies, tests, None, sep)
'momentum': momentum, 'mae': mae, 'rmse': rmse, 'lrate': current_l_w } config_result['results'].append(iteration_result) print(iteration_str.format(j, k, current_l_w, momentum, mae, rmse)) with open('{}_{}.json'.format(config_name, name), 'wt') as res_output: res_output.write(json.dumps(config_result, indent=4)) if __name__ == "__main__": experiment = read_experiment(sys.argv[1])[0] name = experiment['name'] train_path = experiment['train_path'] test_path = experiment['test_path'] sep = experiment['sep'] all_users, all_movies, tests = load_dataset(train_path, test_path, sep, user_based=False) for config in experiment['configs']: run(name, train_path, config, all_users, all_movies, tests, None, sep)