コード例 #1
0
def fit_one(pre=None, ctry=None):
    """
    Similar to gen_params (earlier anyway).  Fits for a single country.
    """
    if pre is None:
        pre = load_pre(ctry)
    by_product = pre.dropna().groupby(level='PRODUCT_NC')
    res = {name: gen_params(group, x0=[2, 1]) for name, group in by_product}
    res = pd.DataFrame(res).T
    for_csv, for_hd5 = opt_dict_format(res, names=['t1', 't2'])
    return (for_csv, for_hd5)
コード例 #2
0
            continue
        #---------------------------------------------------------------------
        # GMM Estimation.
        # Without Weighting
        # res = {name: gen_params(group, [2, 1], name, country=ctry, W=None)
        #        for name, group in by_product}
        #---------------------------------------------------------------------
        # With Weighting
        res = {name: gen_params(
            group, [2, 1], name, country=ctry, W=True, options={'disp': False})
            for name, group in by_product}
        print('Finshed estimation for {}'.format(ctry))
        #---------------------------------------------------------------------
        # Formatting and IO.
        res = pd.DataFrame(res).T
        for_csv, for_hd5 = opt_dict_format(res, names=['t1', 't2'])
        try:
            gmm_results.append('res_' + ctry, for_hd5)
        except:
            with open(base + 'failed_h5.txt', 'a') as f:
                f.write("Missed on {}".format(ctry))
            res.to_csv('/Volumes/HDD/Users/tom/DataStorage/Comext/'
                       'yearly/new_ctry_{}_weighted.csv'.format(ctry))
        # try:
        #     for_csv.to_csv('/Volumes/HDD/Users/tom/DataStorage/Comext/'
        #                    'yearly/new_ctry_{}_weighted.csv'.format(ctry))
        # except:
        #     with open(base + 'failed_csv.txt', 'a') as f:
        #         f.write("Missed on {}".format(ctry))

        m = 'Finshed country {0} in {1}'.format(ctry, datetime.utcnow() - t)