Commit Graph

199 Commits

Author SHA1 Message Date
robcaulk 69b9b35a08 Merge remote-tracking branch 'origin/develop' into feat/add-pytorch-model-support 2023-04-08 13:22:25 +02:00
robcaulk 48d3c8e62e fix model loading from disk bug, improve doc, clarify installation/docker instructions, add a torch tag to the freqairl docker image. Fix seriously outdated prediction_model docstrings 2023-04-08 12:09:53 +02:00
Yinon Polak c137666230 fix imports 2023-04-03 16:03:15 +03:00
Yinon Polak bd3b70293f add pytorch data convertor 2023-04-03 15:19:10 +03:00
Yinon Polak b795a70102 fix config example in pytorch mlp documentation 2023-03-28 14:44:43 +03:00
Yinon Polak 026b6a39a9 bugfix skip test split when empty 2023-03-28 14:40:23 +03:00
Yinon Polak 479aafc331 rename Torch to PyTorch 2023-03-22 17:50:00 +02:00
robcaulk 1ba01746a0 organize pytorch files 2023-03-21 15:09:54 +01:00
Yinon Polak 83a7d888bc type hint init in pytorch mlp classes 2023-03-21 15:19:34 +02:00
Yinon Polak a80afc8f1b add optional target tensor squeezing to pytorch trainer 2023-03-21 13:20:54 +02:00
Yinon Polak 9906e7d646 clean code 2023-03-21 11:23:45 +02:00
Yinon Polak e8f040bfbd add class_name attribute to freqai interface 2023-03-20 20:38:43 +02:00
Yinon Polak a4b617e482 type hints fixes 2023-03-20 20:22:28 +02:00
Yinon Polak c06cd38951 clean code 2023-03-20 19:55:39 +02:00
Yinon Polak 0a55753faf move default attributes of pytorch classifier to initializer,
to prevent mypy from complaining
2023-03-20 19:40:36 +02:00
Yinon Polak 6b4d9f97c1 clean code 2023-03-20 19:28:30 +02:00
Yinon Polak 500c401b75 improve pytorch classifier documentation 2023-03-20 18:41:04 +02:00
Yinon Polak 9aec1ddb17 sort imports 2023-03-20 18:41:04 +02:00
Yinon Polak d98890f32e sort imports 2023-03-20 18:41:04 +02:00
Yinon Polak 54db239175 add pytorch regressor example 2023-03-20 18:41:04 +02:00
Yinon Polak 601c37f862 refactor classifiers class names 2023-03-20 18:41:04 +02:00
Yinon Polak 501e746c52 improve mlp documentation 2023-03-20 18:41:04 +02:00
Yinon Polak d04146d1b1 improve mlp documentation 2023-03-20 18:41:04 +02:00
Yinon Polak ea08931ab3 add mlp documentation 2023-03-20 18:41:04 +02:00
Yinon Polak ddd1b5c0ff modify feedforward net, move layer norm to start of thr block 2023-03-20 18:41:04 +02:00
Yinon Polak fbf7049ac5 sort imports 2023-03-20 18:41:04 +02:00
Yinon Polak 833aaf8e10 create children class to PyTorchClassifier to implement the fit method where we initialize the trainer and model objects 2023-03-20 18:41:04 +02:00
robcaulk fab505be1b cheat flake8 for now until we can refactor save into the model class 2023-03-20 18:41:03 +02:00
Yinon Polak 918889a2bd reduce mlp number of parameters for testing 2023-03-13 20:09:12 +02:00
Yinon Polak d7ea750823 revert to using model_training_parameters 2023-03-13 00:35:51 +02:00
Yinon Polak 0012fe36ca sort imports 2023-03-12 16:16:04 +02:00
Yinon Polak f9fdf1c31b generalize mlp model 2023-03-12 14:31:08 +02:00
Yinon Polak 1cf0e7be24 use one iteration on all test and train data for evaluation 2023-03-12 12:48:15 +02:00
initrv 82cb107520 add tensorboard category 2023-03-12 01:32:55 +03:00
Yinon Polak 8a9f2aedbb improve documentation 2023-03-09 14:55:52 +02:00
Yinon Polak e88a0d5248 convert single quotes to double quotes 2023-03-09 13:29:11 +02:00
Yinon Polak 3081b9402b add documentation 2023-03-09 11:14:54 +02:00
Yinon Polak 1597c3aa89 set class names in IStrategy.set_freqai_targets method, also save class name with model meta data 2023-03-08 18:36:44 +02:00
Yinon Polak 7d26df01b8 fix tensor type hint 2023-03-08 16:17:19 +02:00
Yinon Polak c8296ccb2d sort imports 2023-03-08 16:13:35 +02:00
Yinon Polak 8d60327d60 add missing import 2023-03-08 16:12:47 +02:00
Yinon Polak 04564dc134 add missing import 2023-03-08 16:11:51 +02:00
Yinon Polak 6161b858c4 sort imports 2023-03-08 16:10:25 +02:00
Yinon Polak 1921a07b89 sort imports 2023-03-08 16:08:04 +02:00
Yinon Polak dfbb2e2b35 sort imports 2023-03-08 16:03:36 +02:00
Yinon Polak 1805db2b07 change documentation and small bugfix 2023-03-08 15:38:22 +02:00
Yinon Polak 76fbec0c17 ad multiclass target names encoder to ints 2023-03-08 14:29:38 +02:00
Yinon Polak 8acdd0b47c type hints fixes 2023-03-06 19:14:54 +02:00
Yinon Polak b1ac2bf515 use data loader, add evaluation on epoch 2023-03-06 16:16:45 +02:00
Yinon Polak 751b205618 initial commit 2023-03-05 16:59:24 +02:00