Commit Graph

20051 Commits

Author SHA1 Message Date
Yinon Polak 443263803c unsqueeze target tensor when 1 dimensional 2023-03-21 11:42:05 +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 bf4aa91aab Merge remote-tracking branch 'origin/feat/add-pytorch-model-support' into feat/add-pytorch-model-support
# Conflicts:
#	freqtrade/freqai/base_models/PyTorchModelTrainer.py
#	freqtrade/freqai/prediction_models/PyTorchClassifier.py
#	freqtrade/freqai/prediction_models/PyTorchMLPClassifier.py
#	freqtrade/freqai/prediction_models/PyTorchMLPModel.py
#	tests/freqai/test_freqai_interface.py
2023-03-20 18:44:24 +02:00
Yinon Polak 500c401b75 improve pytorch classifier documentation 2023-03-20 18:41:04 +02:00
Yinon Polak 81a2cbb4eb fix tests 2023-03-20 18:41:04 +02:00
Yinon Polak 0510cf4491 add config params to tests 2023-03-20 18:41:04 +02:00
Yinon Polak 68728409aa add pytorch regressor test 2023-03-20 18:41:04 +02:00
Yinon Polak c00ffcee59 fix pytorch classifier test 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 f659f8e309 remove unused 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 e08d8190ae fix test 2023-03-20 18:41:04 +02:00
Yinon Polak fbf7049ac5 sort imports 2023-03-20 18:41:04 +02:00
Yinon Polak 2a1a8c0e64 fix test 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
Yinon Polak 566346dd87 classifier test - set model file extension 2023-03-20 18:41:03 +02:00
Yinon Polak d0a33d2ee7 fix tests 2023-03-20 18:41:03 +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 2f386913ac refactor classifiers class names 2023-03-20 11:54:17 +02:00
Yinon Polak 1c11a5f048 improve mlp documentation 2023-03-19 18:10:57 +02:00
Yinon Polak 903a1dc3e5 improve mlp documentation 2023-03-19 18:04:01 +02:00
Yinon Polak 6f9a8a089c add mlp documentation 2023-03-19 17:45:30 +02:00
Yinon Polak 8bee499328 modify feedforward net, move layer norm to start of thr block 2023-03-19 17:03:36 +02:00
Yinon Polak 719faab4b8 fix test 2023-03-19 15:21:34 +02:00
Yinon Polak 9f477aa3c9 sort imports 2023-03-19 15:09:50 +02:00
Yinon Polak 61ac36c576 fix test 2023-03-19 14:49:12 +02:00
Yinon Polak 366c148c10 create children class to PyTorchClassifier to implement the fit method where we initialize the trainer and model objects 2023-03-19 14:38:49 +02:00
Yinon Polak a49f62eecb classifier test - set model file extension 2023-03-18 20:51:30 +02:00
Yinon Polak fab9ff1294 fix tests 2023-03-18 15:27:38 +02:00
Yinon Polak 1c91b4427b Merge remote-tracking branch 'origin/feat/add-pytorch-model-support' into feat/add-pytorch-model-support 2023-03-18 14:14:38 +02:00
Yinon Polak 244662b1a4 set class names attribute in the general classifier testing strategy 2023-03-18 14:12:31 +02:00
robcaulk 4550447409 cheat flake8 for now until we can refactor save into the model class 2023-03-14 21:13:30 +01:00
Yinon Polak 366740885a reduce mlp number of parameters for testing 2023-03-13 20:18:26 +02:00
Yinon Polak 918889a2bd reduce mlp number of parameters for testing 2023-03-13 20:09:12 +02:00
Yinon Polak 9c8c30b0e8 add test 2023-03-13 17:17:00 +02:00
Yinon Polak d7ea750823 revert to using model_training_parameters 2023-03-13 00:35:51 +02:00
Yinon Polak b6096efadd logging change 2023-03-13 00:35:14 +02:00
Yinon Polak b927c9dc01 remove train loss calculation from estimate_loss 2023-03-13 00:17:34 +02:00
Yinon Polak 523a58d3d6 simplify statement for pytorch file_type extension 2023-03-13 00:16:44 +02:00