In [1]:
import Loader
import Features

In [2]:
#create a single candidate object (K2), with ID of 2044 (arbitrary)
dat=Loader.Candidate(2044,'ktwo204490599c02_lpd_LC.txt',observatory='K2')
dat


Out[2]:
<Loader.Loader.Candidate at 0x111982dd0>

In [3]:
#set up a featureset object
feat=Features.Featureset(dat,useflatten=False)
feat


Out[3]:
<Features.Featureset.Featureset at 0x1119ee710>

In [4]:
#calculate some features
feat.CalcFeatures(featuredict={'SPhot_max':[],'Contrast':[],'LSPeriod':[1],'LSAmp':[1],'F8':[]})


Features/lomb.py:151: VisibleDeprecationWarning: using a non-integer number instead of an integer will result in an error in the future
  wk1 = wk1[1:nout+1]
Features/lomb.py:152: VisibleDeprecationWarning: using a non-integer number instead of an integer will result in an error in the future
  wk2 = wk2[1:nout+1]
Calculating LS Period first

In [5]:
#show current set of calculates features (note, LSPeriod0 is present as it was calculated by the SPhot_max function)
feat.features


Out[5]:
{'Contrast': 1.2356388244476855,
 'F8': 0.00019326222177056034,
 'LSAmp1': 0.37521055048983798,
 'LSPeriod0': 1.9599740639195893,
 'LSPeriod1': 0.96690392473083342,
 'SPhot_max': 0.00041158940272038896}

In [6]:
#calculate some more features
feat.CalcFeatures(featuredict={'LSPeriod':[3],'LSAmp':[3]})
feat.features


Out[6]:
{'Contrast': 1.2356388244476855,
 'F8': 0.00019326222177056034,
 'LSAmp1': 0.37521055048983798,
 'LSAmp3': 0.24805572283418356,
 'LSPeriod0': 1.9599740639195893,
 'LSPeriod1': 0.96690392473083342,
 'LSPeriod3': 0.9914862279019564,
 'SPhot_max': 0.00041158940272038896}

In [7]:
#now let's use some tsfresh features to save us some coding
feat.CalcFeatures(featuredict={'TSFresh':['flux__median']})


Feature Extraction: 100%|██████████| 1/1 [00:18<00:00, 18.04s/it]

In [9]:
#using more tsfresh features takes no more time
feat.CalcFeatures(featuredict={'TSFresh':['flux__time_reversal_asymmetry_statistic__lag_2']})
feat.features


Out[9]:
{'Contrast': 1.2356388244476855,
 'F8': 0.00019326222177056034,
 'LSAmp1': 0.37521055048983798,
 'LSAmp3': 0.24805572283418356,
 'LSPeriod0': 1.9599740639195893,
 'LSPeriod1': 0.96690392473083342,
 'LSPeriod3': 0.9914862279019564,
 'SPhot_max': 0.00041158940272038896,
 'TSFreshflux__median': 0.99996021664223045,
 'TSFreshflux__time_reversal_asymmetry_statistic__lag_2': -3.3240814610343844e-07}

In [ ]: