Skip to content

Updating/Generalizing Light Curve to Tensor Loading #397

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
malinadesai wants to merge 7 commits into
base: main
Choose a base branch
from
Open

Updating/Generalizing Light Curve to Tensor Loading #397

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
69d7556
Update dataprocessing.py
malinadesai Jun 26, 2025
d707cff
Create lightcurves_to_tensors.py
malinadesai Jun 26, 2025
c228a34
Create __init__.py
malinadesai Jun 26, 2025
cea8c0d
Update embedding.py
malinadesai Jun 30, 2025
2f398af
Update dataprocessing.py
malinadesai Jul 7, 2025
08c68b6
Update lightcurves_to_tensors.py
malinadesai Jul 7, 2025
eedb6e6
use brendan's injection.py for rubin ToOs
malinadesai Jul 9, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
68 changes: 11 additions & 57 deletions nmma/em/injection.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -326,68 +326,23 @@ def create_light_curve_data(
sim.to_csv(args.outdir + "/too.csv", index=False)

if rubin_ToO:
print("Using rubin observing strategy.")
start = tmin + tc
if args.rubin_ToO_type == "platinum":
# platinum means 90% GW skymap <30 sq deg
# I made this name up, this is the gold strategy for an event similar to GW170817 (close and well localized)
# Three observations Night0 with grizy filters
# One scan Night 1,2,3 w/ same filters
if args.rubin_ToO_type == "BNS":
strategy = [
[1 / 24.0, ["ps1__g", "ps1__r", "ps1__i", "ps1__z", "ps1__y"]],
[2 / 24.0, ["ps1__g", "ps1__r", "ps1__i", "ps1__z", "ps1__y"]],
[4 / 24.0, ["ps1__g", "ps1__r", "ps1__i", "ps1__z", "ps1__y"]],
[1.0, ["ps1__g", "ps1__r", "ps1__i", "ps1__z", "ps1__y"]],
[2.0, ["ps1__g", "ps1__r", "ps1__i", "ps1__z", "ps1__y"]],
[3.0, ["ps1__g", "ps1__r", "ps1__i", "ps1__z", "ps1__y"]],
[1 / 24.0, ["sdssu", "ps1__g", "ps1__r", "ps1__i", "ps1__z", "ps1__y"]],
[2 / 24.0, ["sdssu", "ps1__g", "ps1__r", "ps1__i", "ps1__z", "ps1__y"]],
[4 / 24.0, ["sdssu", "ps1__g", "ps1__r", "ps1__i", "ps1__z", "ps1__y"]],
[1.0, ["ps1__g", "ps1__z", "ps1__i"]],
]
elif args.rubin_ToO_type == "gold":
# gold means 90% GW skymap <100 sq deg
# Three pointings Night 0 with gri (possibly grz if more sensitive to KNe)
# One scan Night 1,2,3 w/ r+i
elif args.rubin_ToO_type == "NSBH":
strategy = [
[1 / 24.0, ["ps1__g", "ps1__r", "ps1__i"]],
[2 / 24.0, ["ps1__g", "ps1__r", "ps1__i"]],
[4 / 24.0, ["ps1__g", "ps1__r", "ps1__i"]],
[1.0, ["ps1__r", "ps1__i"]],
[2.0, ["ps1__r", "ps1__i"]],
[3.0, ["ps1__r", "ps1__i"]],
]
elif args.rubin_ToO_type == "gold_z":
# gold means 90% GW skymap <100 sq deg
# Three pointings Night 0 with gri (possibly grz if more sensitive to KNe)
# One scan Night 1,2,3 w/ r+i
strategy = [
[1 / 24.0, ["ps1__g", "ps1__r", "ps1__z"]],
[2 / 24.0, ["ps1__g", "ps1__r", "ps1__z"]],
[4 / 24.0, ["ps1__g", "ps1__r", "ps1__z"]],
[1.0, ["ps1__r", "ps1__i"]],
[2.0, ["ps1__r", "ps1__i"]],
[3.0, ["ps1__r", "ps1__i"]],
]
elif args.rubin_ToO_type == "silver":
# silver means 90% GW skymap <500 sq deg
# One scan Night 0 w/ g+i or g+z
# One scan each Night 1,2,3 w/ same filters
strategy = [
[1 / 24.0, ["ps1__g", "ps1__i"]],
[1.0, ["ps1__g", "ps1__i"]],
[2.0, ["ps1__g", "ps1__i"]],
[3.0, ["ps1__g", "ps1__i"]],
]
elif args.rubin_ToO_type == "silver_z":
# silver means 90% GW skymap <500 sq deg
# One scan Night 0 w/ g+i or g+z
# One scan each Night 1,2,3 w/ same filters
strategy = [
[1 / 24.0, ["ps1__g", "ps1__z"]],
[1.0, ["ps1__g", "ps1__z"]],
[2.0, ["ps1__g", "ps1__z"]],
[3.0, ["ps1__g", "ps1__z"]],
[1 / 24.0, ["sdssu", "ps1__g", "ps1__r", "ps1__i", "ps1__z", "ps1__y"]],
[4 / 24.0, ["sdssu", "ps1__g", "ps1__r", "ps1__i", "ps1__z", "ps1__y"]],
[1.0, ["sdssu", "ps1__g", "ps1__r", "ps1__i", "ps1__z", "ps1__y"]],
[2.0, ["ps1__g", "ps1__z", "ps1__i"]],
]
else:
raise ValueError("args.rubin_ToO_type should be either platinum, gold, or silver")
# took type names from Rubin 2024 Workshop write-up
raise ValueError("args.rubin_ToO_type should be either BNS or NSBH")

mjds, passbands = [], []
sim = pd.DataFrame()
Expand All @@ -414,7 +369,6 @@ def create_light_curve_data(
assume_sorted=True,
)
times = group["mjd"].tolist()
#print("The times of observation are: ", times)
data_per_filt = np.vstack([times, lc(times), lcerr(times)]).T
data[filt] = data_per_filt
passbands_to_keep.append(filt)
Expand Down
1 change: 1 addition & 0 deletions nmma/mlmodel/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@

Loading
Loading