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