def test_module_loading(self): pyjags.load_module('basemod') pyjags.load_module('bugs') pyjags.load_module('lecuyer') self.assertEqual( ['basemod', 'bugs', 'lecuyer'], pyjags.list_modules())
import argparse import os import pickle import matplotlib.pyplot as plt import numpy as np import pyjags import xarray as xr from utils import * # Load additional JAGS module pyjags.load_module('glm') plt.style.use('ggplot') DIAGNOSIS = False DIR = '../results/' OUTPUT_DIR = '../results/' def main(): experiment_name = '%s_%s_budget%d' % (args.dataset, args.attribute, args.budget) path = OUTPUT_DIR + experiment_name if not os.path.exists(path): os.makedirs(path) dataset = Dataset.load_from_file(DIR + "%s_%s_scores_remapped.csv" % (args.dataset, args.attribute), args.dataset) print("\n\n\n================%s================" % dataset.dataset_name) dataset.shuffle(random_state=args.run_id, attribute=args.attribute)
def test_module_loading(self): pyjags.load_module('basemod') pyjags.load_module('bugs') pyjags.load_module('lecuyer') self.assertEqual(['basemod', 'bugs', 'lecuyer'], pyjags.list_modules())