easonabc-public
/
paia-hw5
1
0
Fork 1
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

remove check every feature

This commit is contained in:
Eason 2024-06-12 23:24:22 +08:00
parent 4aec307596
commit 336594dfa4
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tank-rust/src/dqn/feature.rs
View File

@ -124,7 +124,7 @@ impl<'a> Info<'a> {
let angle = self.player.get_angle(); let angle = self.player.get_angle();
let gun_angle = self.player.get_gun_angle(); let gun_angle = self.player.get_gun_angle();
let feature=[ [
normalize_angle(target.angle - angle).tanh(), normalize_angle(target.angle - angle).tanh(),
(wall.distance - target.distance).tanh(), (wall.distance - target.distance).tanh(),
(self.player.power as f32).tanh(), (self.player.power as f32).tanh(),
@ -132,11 +132,7 @@ impl<'a> Info<'a> {
(emeny.distance + 1.0).log2(), (emeny.distance + 1.0).log2(),
normalize_angle(emeny.angle - gun_angle).tanh(), normalize_angle(emeny.angle - gun_angle).tanh(),
normalize_angle(wall.angle - gun_angle).tanh(), normalize_angle(wall.angle - gun_angle).tanh(),
]; ]
for f in feature.iter() {
assert!(!f.is_nan(),"feature is nan: {:?}", feature);
}
feature
} }
pub fn into_feature_tensor<B: Backend>(&self, device: &B::Device) -> Tensor<B, 1> { pub fn into_feature_tensor<B: Backend>(&self, device: &B::Device) -> Tensor<B, 1> {
let feature = self.into_feature(); let feature = self.into_feature();