Yinon Polak
|
97339e14cf
|
round up divisions in calc_n_epochs
|
2023-03-21 12:29:05 +02:00 |
|
Yinon Polak
|
443263803c
|
unsqueeze target tensor when 1 dimensional
|
2023-03-21 11:42:05 +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
|
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
|
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
|
1cf0e7be24
|
use one iteration on all test and train data for evaluation
|
2023-03-12 12:48:15 +02:00 |
|
Yinon Polak
|
e88a0d5248
|
convert single quotes to double quotes
|
2023-03-09 13:29:11 +02:00 |
|
Yinon Polak
|
2ef11faba7
|
reformat documentation
|
2023-03-09 13:25:20 +02:00 |
|
Yinon Polak
|
c9eee2944b
|
reformat documentation
|
2023-03-09 13:01:04 +02:00 |
|
Yinon Polak
|
6f962362f2
|
expand pytorch trainer documentation
|
2023-03-09 12:45:46 +02:00 |
|
Yinon Polak
|
ba5de0cd00
|
add documentation
|
2023-03-09 11:21:10 +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
|
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
|
4241bff32a
|
type hints fixes
|
2023-03-06 20:15:36 +02:00 |
|
Yinon Polak
|
5dd60eda36
|
type hints fixes
|
2023-03-06 19:37:08 +02:00 |
|
Yinon Polak
|
8acdd0b47c
|
type hints fixes
|
2023-03-06 19:14:54 +02:00 |
|
Yinon Polak
|
e6e747bcd8
|
reformat code
|
2023-03-06 17:50:02 +02:00 |
|
Yinon Polak
|
348a08f1c4
|
add todo - currently assuming class labels are strings ['0.0', '1.0' .. n_classes]. need to resolve it per ClassifierModel
|
2023-03-06 16:41:47 +02:00 |
|
Yinon Polak
|
b1ac2bf515
|
use data loader, add evaluation on epoch
|
2023-03-06 16:16:45 +02:00 |
|