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())