Пример #1
0
        vary_planet = (['period', 'rprs', 'impact', 'T0'])

        fitT.free_parameters(vary_star, vary_planet)
        fitT.do_fit()

        #print(fitT.fitresultstellar.items())
        #print(fitT.fitresultplanets.items())

        bestFstellar = fitT.fitresultstellar.items()
        bestFrho = bestFstellar[0][1]  #Best Fit Rho

        bestFplanet = fitT.fitresultplanets.items()
        bestFperiod = bestFplanet[0][1]['period']  #Best Fit Period
        bestFrprs = bestFplanet[0][1]['rprs']  #Best Fit rprs

        fitT.print_results()

        #save figure
        #fig = ktransit.plot_results(time,fluxDetrend,fitT.transitmodel)
        #fig.savefig('/Users/sheilasagear/OneDrive/K2_Research/CorrectedLC_EPICID/ktransitfits_untrendy/' + str(targetname) + 'fitPer' + str(p) + 'Rprs' + str(r) + '.png')

        #if not os.path.exists("/Users/sheilasagear/OneDrive/K2_Research/CorrectedLC_EPICID/EPIC" + str(targetname)):
        #    os.makedirs("/Users/sheilasagear/OneDrive/K2_Research/CorrectedLC_EPICID/EPIC" + str(targetname))
        """        
        pylab.cla()
        fig, ax = plt.subplots(1, 1, figsize=[11,5])
        ax.scatter(time, fluxDetrend, color='k', s=2)
        ax.plot(time, fitT.transitmodel, color='mediumaquamarine')
        ax.set_ylim(-0.07,0.07)
        ax.set_xlabel('Time (BJD - 2450000)')
        ax.set_title('EPIC ' + str(targetname))
Пример #2
0
                      esinw=0.0)  #simult, sff

fitT.add_data(time=t, flux=np.array(y))

vary_star = ['zpt']  #, 'ld1', 'ld2']      # free stellar parameters
vary_planet = ([
    'period',  # free planetary parameters
    'T0',
    'impact',  # 'esinw', 'ecosw',
    'rprs'
])  # free planet parameters are the same for every planet you model

fitT.free_parameters(vary_star, vary_planet)
fitT.do_fit()  # run the fitting

fitT.print_results()  # print some results
res = fitT.fitresultplanets
res2 = fitT.fitresultstellar

print("transit model shape:", fitT.transitmodel.shape)


class TransitModel(Model):
    parameter_names = ("log_ror", "log_rho", "log_T0", "log_per", "log_imp"
                       )  #, "ecosw", "esinw")

    def get_value(self, t):

        #make transit model
        M = ktransit.LCModel()
        M.add_star(