Exemplo n.º 1
0
    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)
Exemplo n.º 3
0
    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())