File size: 792 Bytes
59bd924 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
# scripts/make_toy_sensor.py
import numpy as np, pandas as pd
rng = np.random.default_rng(1)
rows = []
for subj in range(3):
T = 400
acc_x = rng.normal(0, 0.8, size=T)
acc_y = rng.normal(0, 0.8, size=T)
acc_z = rng.normal(0, 1.2, size=T)
# 伪标签：0=rest,1=walk,2=run
activity = np.zeros(T, dtype=object)
activity[100:250] = "walk"
activity[250:350] = "run"
# 奖励表达式将惩罚加速度幅度（越平稳越好）
for t in range(T):
rows.append(dict(
subject=f"S{subj}", frame=t,
acc_x=acc_x[t], acc_y=acc_y[t], acc_z=acc_z[t],
activity=activity[t]
))
df = pd.DataFrame(rows)
df.to_csv("./data/toy_sensor.csv", index=False)
print("[OK] wrote ./data/toy_sensor.csv with", len(df), "rows")
|