diff options
| author | Alexey Donskov <voxnod@gmail.com> | 2025-03-22 19:59:25 +0300 |
|---|---|---|
| committer | Gleb Popov <arrowd@FreeBSD.org> | 2025-03-28 15:37:25 +0300 |
| commit | 8f2823654667cc746695a83126e15c3172c6261d (patch) | |
| tree | 30287c25edd29252f9037d8692f6831a8d2cb79b /science/linux-ai-ml-env/files/patch-time__sequence__prediction_train.py | |
| parent | 7ed0bfc22fba833d844ff4c2e0b1a3b9f298c34b (diff) | |
science/linux-ai-ml-env: Linux Python environment for running AI and ML stuff
Sponsored by: Future Crew, LLC
Pull Request: https://github.com/freebsd/freebsd-ports/pull/361
Diffstat (limited to 'science/linux-ai-ml-env/files/patch-time__sequence__prediction_train.py')
| -rw-r--r-- | science/linux-ai-ml-env/files/patch-time__sequence__prediction_train.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/science/linux-ai-ml-env/files/patch-time__sequence__prediction_train.py b/science/linux-ai-ml-env/files/patch-time__sequence__prediction_train.py new file mode 100644 index 000000000000..66a1b46c198b --- /dev/null +++ b/science/linux-ai-ml-env/files/patch-time__sequence__prediction_train.py @@ -0,0 +1,11 @@ +--- time_sequence_prediction/train.py.orig 2025-02-09 16:56:58 UTC ++++ time_sequence_prediction/train.py +@@ -44,7 +44,7 @@ if __name__ == '__main__': + np.random.seed(0) + torch.manual_seed(0) + # load data and make training set +- data = torch.load('traindata.pt') ++ data = torch.load('traindata.pt', weights_only=False) + input = torch.from_numpy(data[3:, :-1]) + target = torch.from_numpy(data[3:, 1:]) + test_input = torch.from_numpy(data[:3, :-1]) |