11import numpy as np
2- import functools
32import json
43import sys
5- import os .path
6- import ad
7- import multiprocessing
8- import os
9- import scipy .stats
10- import concurrent .futures as futures
11- import itertools as it
12- from scipy .stats .mstats import mquantiles
134
145from . import estimation_tools , _smcpp , util , logging , spline , data_filter , beta_de
156import smcpp .defaults
16- from .contig import Contig
17- from .model import SMCModel , SMCTwoPopulationModel , PiecewiseModel
7+ from .model import SMCModel , SMCTwoPopulationModel
188from smcpp .optimize .optimizers import SMCPPOptimizer , TwoPopulationOptimizer
199from smcpp .optimize .plugins import analysis_saver , parameter_optimizer
2010
@@ -36,7 +26,6 @@ def __init__(self, files, args):
3626 self ._rho = 2 * self ._N0 * args .r
3727 else :
3828 self ._rho = self ._theta
39- self ._cM = 1e-2 / (self ._rho / (2 * self ._N0 ))
4029 assert np .all (np .isfinite ([self ._rho , self ._theta ]))
4130 logger .info ("rho: %f" , self ._rho )
4231 self ._penalty = 0.
@@ -110,7 +99,6 @@ def _init_inference_manager(self, polarization_error, hs):
11099 im .alpha = self ._alpha = 1
111100 im .polarization_error = polarization_error
112101 self ._ims [pid ] = im
113- self ._max_n = np .max (list (map (sum , max_n .values ())), axis = 0 )
114102
115103 # @property
116104 # def _data(self):
@@ -236,7 +224,7 @@ def __init__(self, files, args):
236224
237225 logger .debug ("hidden states in coalescent scaling: %s" , hs )
238226
239- self ._init_model (self . _N0 , args .spline )
227+ self ._init_model (args .spline )
240228 self ._init_inference_manager (args .polarization_error , self ._hidden_states )
241229 self .alpha = args .w
242230 self ._model [:] = np .log (NeN0 )
@@ -247,7 +235,7 @@ def __init__(self, files, args):
247235 self ._init_regularization (args )
248236
249237
250- def _init_model (self , N0 , spline_class ):
238+ def _init_model (self , spline_class ):
251239 ## Initialize model
252240 logger .debug ("knots in coalescent scaling:\n %s" , str (self ._knots ))
253241 spline_class = {"cubic" : spline .CubicSpline ,
0 commit comments