-
NOTE: only trained on poly 1 for now
-
Program uses burn LSTM + Linear modules to predict the next number in a linear sequnece
-
Trained on and works best with 5 preceeding inputs
-
Yes.. I know this is a pointless program. I code for the LEARNINGSSSSSS
- Requires dedicated GPU
cargo run --bin train --release
Training takes about 5 minutes with an RTX 4060;
cargo run --bin infer 18,6,-6,-18,-30