alu/tra-analysis
1
0
Fork 0
mirror of https://github.com/titanscouting/tra-analysis.git synced 2025-10-23 17:49:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

analysis.py v 1.1.13.006

Browse Source 
regression.py v 1.0.0.003
analysis pkg v 1.0.0.8
This commit is contained in:
ltcptgeneral
2020-03-08 16:48:19 -05:00
parent 40e5899972
commit 04141bbec8
12 changed files with 70 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
analysis-master/analysis/regression.py
View File

@@ -5,19 +5,22 @@
# this module is cuda-optimized and vectorized (except for one small part)
# setup:
__version__ = "1.0.0.003"
__version__ = "1.0.0.004"
# changelog should be viewed using print(analysis.regression.__changelog__)
__changelog__ = """
1.0.0.003:
- bug fixes
1.0.0.002:
-Added more parameters to log, exponential, polynomial
-Added SigmoidalRegKernelArthur, because Arthur apparently needs
to train the scaling and shifting of sigmoids
1.0.0.001:
-initial release, with linear, log, exponential, polynomial, and sigmoid kernels
-already vectorized (except for polynomial generation) and CUDA-optimized
1.0.0.004:
- bug fixes
- fixed changelog
1.0.0.003:
- bug fixes
1.0.0.002:
-Added more parameters to log, exponential, polynomial
-Added SigmoidalRegKernelArthur, because Arthur apparently needs
to train the scaling and shifting of sigmoids
1.0.0.001:
-initial release, with linear, log, exponential, polynomial, and sigmoid kernels
-already vectorized (except for polynomial generation) and CUDA-optimized
"""
__author__ = (
@@ -40,6 +43,8 @@ __all__ = [
'CustomTrain'
]
import torch
global device
device = "cuda:0" if torch.torch.cuda.is_available() else "cpu"