second commit
This commit is contained in:
parent
24e44bbd35
commit
34cd306289
6
main.py
6
main.py
@ -4,10 +4,6 @@ import numpy as np
|
||||
import pandas as pd
|
||||
from torch.autograd import Variable
|
||||
|
||||
train = 'train/train.tsv'
|
||||
in_file = 'test-A/in.tsv'
|
||||
out_file = 'test-A/out.tsv'
|
||||
learning_rate = 0.00001
|
||||
|
||||
|
||||
class linearRegression(nn.Module):
|
||||
@ -37,7 +33,7 @@ if torch.cuda.is_available():
|
||||
model.cuda()
|
||||
|
||||
criterion = nn.MSELoss()
|
||||
optimizer = torch.optim.SGD(model.parameters(), lr=learning_rate)
|
||||
optimizer = torch.optim.SGD(model.parameters(), lr=0.00001)
|
||||
|
||||
|
||||
|
||||
|
836
test-A/out.tsv
836
test-A/out.tsv
@ -1,418 +1,418 @@
|
||||
40558.543
|
||||
41873.387
|
||||
33467.543
|
||||
29948.207
|
||||
29025.854
|
||||
60778.36
|
||||
29117.434
|
||||
52339.81
|
||||
49808.242
|
||||
31969.535
|
||||
39904.39
|
||||
40565.086
|
||||
31668.625
|
||||
24865.451
|
||||
31406.963
|
||||
32061.115
|
||||
59273.81
|
||||
34677.723
|
||||
19632.24
|
||||
39858.598
|
||||
48414.9
|
||||
20286.393
|
||||
32715.266
|
||||
25519.602
|
||||
56918.867
|
||||
31936.828
|
||||
41644.434
|
||||
35887.902
|
||||
17872.572
|
||||
31930.285
|
||||
65422.832
|
||||
41873.387
|
||||
85047.375
|
||||
22831.04
|
||||
28790.36
|
||||
47499.086
|
||||
33369.418
|
||||
36640.176
|
||||
28136.207
|
||||
46308.53
|
||||
41677.14
|
||||
35914.066
|
||||
23884.225
|
||||
39256.78
|
||||
24211.299
|
||||
33068.51
|
||||
39976.348
|
||||
23380.527
|
||||
20090.146
|
||||
60843.773
|
||||
38622.254
|
||||
18847.258
|
||||
25454.188
|
||||
40434.254
|
||||
31406.963
|
||||
42540.617
|
||||
40179.133
|
||||
23517.898
|
||||
31760.207
|
||||
27220.395
|
||||
58030.92
|
||||
34023.57
|
||||
32061.115
|
||||
54302.26
|
||||
25172.902
|
||||
35331.875
|
||||
53713.523
|
||||
34926.3
|
||||
31406.963
|
||||
25650.434
|
||||
22248.846
|
||||
40565.086
|
||||
40892.16
|
||||
37856.9
|
||||
29902.416
|
||||
40892.16
|
||||
27612.887
|
||||
28790.36
|
||||
31936.828
|
||||
35986.023
|
||||
62845.477
|
||||
55525.527
|
||||
43423.727
|
||||
41546.312
|
||||
20940.543
|
||||
35547.746
|
||||
27534.389
|
||||
40238.008
|
||||
31799.453
|
||||
24865.451
|
||||
36110.312
|
||||
22883.373
|
||||
40172.594
|
||||
31406.963
|
||||
30752.812
|
||||
90280.586
|
||||
67712.37
|
||||
41022.992
|
||||
24930.865
|
||||
23360.902
|
||||
34677.723
|
||||
32715.266
|
||||
34023.57
|
||||
38602.63
|
||||
27678.3
|
||||
30164.076
|
||||
41219.234
|
||||
58802.82
|
||||
58456.12
|
||||
28136.207
|
||||
71964.34
|
||||
36018.73
|
||||
19730.363
|
||||
26173.754
|
||||
59862.547
|
||||
19697.656
|
||||
28195.08
|
||||
34873.97
|
||||
29444.51
|
||||
26762.49
|
||||
30229.492
|
||||
28424.033
|
||||
29974.373
|
||||
32401.273
|
||||
46020.703
|
||||
19501.41
|
||||
25912.092
|
||||
32761.057
|
||||
38511.05
|
||||
39256.78
|
||||
38733.46
|
||||
36980.336
|
||||
24865.451
|
||||
25519.602
|
||||
41415.48
|
||||
21790.94
|
||||
30098.662
|
||||
32715.266
|
||||
34023.57
|
||||
59535.473
|
||||
35325.332
|
||||
22327.344
|
||||
41742.555
|
||||
42135.047
|
||||
56918.867
|
||||
28790.36
|
||||
29052.02
|
||||
71964.34
|
||||
41219.234
|
||||
32904.973
|
||||
20168.645
|
||||
28136.207
|
||||
22942.246
|
||||
43835.84
|
||||
24211.299
|
||||
30752.812
|
||||
90928.195
|
||||
82914.836
|
||||
32414.357
|
||||
66534.89
|
||||
43835.84
|
||||
26173.754
|
||||
31406.963
|
||||
37130.79
|
||||
44045.168
|
||||
13110.354
|
||||
38844.664
|
||||
36640.176
|
||||
34677.723
|
||||
39250.242
|
||||
18781.844
|
||||
46648.69
|
||||
33022.72
|
||||
25519.602
|
||||
24374.838
|
||||
28829.61
|
||||
30752.812
|
||||
30229.492
|
||||
44489.992
|
||||
25068.238
|
||||
24865.451
|
||||
42527.54
|
||||
48414.9
|
||||
34808.555
|
||||
25218.693
|
||||
26304.584
|
||||
46799.145
|
||||
91588.88
|
||||
34677.723
|
||||
43443.35
|
||||
37294.33
|
||||
33977.78
|
||||
31406.963
|
||||
75235.1
|
||||
20888.21
|
||||
38668.043
|
||||
28136.207
|
||||
32715.266
|
||||
28855.773
|
||||
34023.57
|
||||
79290.836
|
||||
38602.63
|
||||
26749.408
|
||||
30949.057
|
||||
49069.05
|
||||
35312.25
|
||||
16793.223
|
||||
27370.852
|
||||
26592.41
|
||||
178329.36
|
||||
32015.324
|
||||
32061.115
|
||||
27416.643
|
||||
44097.504
|
||||
26173.754
|
||||
62152.08
|
||||
48414.9
|
||||
16793.223
|
||||
21333.033
|
||||
94205.49
|
||||
32061.115
|
||||
32172.32
|
||||
27482.057
|
||||
19730.363
|
||||
34677.723
|
||||
26239.168
|
||||
36842.965
|
||||
74580.95
|
||||
28790.36
|
||||
28790.36
|
||||
32061.115
|
||||
46079.58
|
||||
24538.375
|
||||
28136.207
|
||||
46799.145
|
||||
35887.902
|
||||
51234.29
|
||||
141304.39
|
||||
36640.176
|
||||
34088.984
|
||||
39910.934
|
||||
43835.84
|
||||
32061.115
|
||||
29490.3
|
||||
41873.387
|
||||
39453.027
|
||||
20286.393
|
||||
24865.451
|
||||
28240.871
|
||||
54498.508
|
||||
75693.01
|
||||
20286.393
|
||||
33441.375
|
||||
26958.736
|
||||
48414.9
|
||||
52470.637
|
||||
41480.9
|
||||
49644.703
|
||||
35887.902
|
||||
16426.898
|
||||
26383.082
|
||||
26834.445
|
||||
50724.055
|
||||
40172.594
|
||||
38144.723
|
||||
28692.234
|
||||
30621.982
|
||||
43835.84
|
||||
49723.203
|
||||
44489.992
|
||||
43835.84
|
||||
28672.611
|
||||
57573.02
|
||||
49742.824
|
||||
53975.188
|
||||
37948.48
|
||||
23426.318
|
||||
35789.78
|
||||
41873.387
|
||||
31406.963
|
||||
32715.266
|
||||
49069.05
|
||||
43181.688
|
||||
32826.473
|
||||
16793.223
|
||||
30752.812
|
||||
30752.812
|
||||
42462.125
|
||||
31406.963
|
||||
43835.84
|
||||
24237.467
|
||||
21476.947
|
||||
35325.332
|
||||
47603.754
|
||||
63002.477
|
||||
31668.625
|
||||
51685.656
|
||||
24865.451
|
||||
31406.963
|
||||
26500.828
|
||||
41480.9
|
||||
23164.658
|
||||
26108.34
|
||||
27586.72
|
||||
27024.15
|
||||
32061.115
|
||||
59535.473
|
||||
22183.432
|
||||
37883.066
|
||||
32061.115
|
||||
28371.703
|
||||
35331.875
|
||||
42527.54
|
||||
25166.36
|
||||
30621.982
|
||||
35554.285
|
||||
58881.32
|
||||
55872.227
|
||||
32636.77
|
||||
31406.963
|
||||
20135.938
|
||||
30249.115
|
||||
32061.115
|
||||
33369.418
|
||||
26965.275
|
||||
22968.41
|
||||
34677.723
|
||||
43835.84
|
||||
40107.176
|
||||
59273.81
|
||||
42854.613
|
||||
24211.299
|
||||
56526.38
|
||||
44444.203
|
||||
48087.824
|
||||
22902.996
|
||||
27612.887
|
||||
34088.984
|
||||
43508.766
|
||||
20135.938
|
||||
39315.656
|
||||
28718.402
|
||||
37752.234
|
||||
34455.312
|
||||
66534.89
|
||||
51554.83
|
||||
30621.982
|
||||
46799.145
|
||||
35586.992
|
||||
68451.555
|
||||
27482.057
|
||||
40434.254
|
||||
22902.996
|
||||
52339.81
|
||||
13090.729
|
||||
31132.22
|
||||
67018.96
|
||||
26173.754
|
||||
28790.36
|
||||
98130.4
|
||||
30667.771
|
||||
23125.408
|
||||
33886.2
|
||||
37974.645
|
||||
43835.84
|
||||
31727.5
|
||||
44751.652
|
||||
77851.71
|
||||
31406.963
|
||||
34873.97
|
||||
31406.963
|
||||
93911.125
|
||||
23295.486
|
||||
17081.05
|
||||
29437.969
|
||||
51685.656
|
||||
32715.266
|
||||
27200.771
|
||||
34056.277
|
||||
26298.041
|
||||
74077.26
|
||||
23949.639
|
||||
80926.22
|
||||
37771.86
|
||||
32054.574
|
||||
57815.055
|
||||
31570.502
|
||||
28083.875
|
||||
44032.09
|
||||
44032.09
|
||||
66534.89
|
||||
45562.8
|
||||
29248.266
|
||||
32636.77
|
||||
35887.902
|
||||
30366.863
|
||||
32061.115
|
||||
22144.182
|
||||
31073.348
|
||||
39256.78
|
||||
32362.023
|
||||
49873.66
|
||||
34965.547
|
||||
34965.547
|
||||
51685.656
|
||||
26500.828
|
||||
56931.95
|
||||
93093.44
|
||||
44738.566
|
||||
38112.016
|
||||
32427.441
|
||||
19651.865
|
||||
23544.064
|
||||
22883.373
|
||||
32100.365
|
||||
50187.65
|
||||
47518.71
|
||||
43077.023
|
||||
40591.41
|
||||
41907.324
|
||||
33494.63
|
||||
29972.426
|
||||
29049.322
|
||||
60827.71
|
||||
29140.977
|
||||
52382.277
|
||||
49848.645
|
||||
31995.402
|
||||
39936.727
|
||||
40597.957
|
||||
31694.248
|
||||
24885.527
|
||||
31432.373
|
||||
32087.057
|
||||
59321.934
|
||||
34705.793
|
||||
19648.05
|
||||
39890.895
|
||||
48454.168
|
||||
20302.736
|
||||
32741.742
|
||||
25540.213
|
||||
56965.07
|
||||
31962.668
|
||||
41678.188
|
||||
35916.96
|
||||
17886.95
|
||||
31956.121
|
||||
65475.97
|
||||
41907.324
|
||||
85116.51
|
||||
22849.457
|
||||
28813.635
|
||||
47537.61
|
||||
33396.426
|
||||
36669.848
|
||||
28158.951
|
||||
46346.086
|
||||
41710.918
|
||||
35943.15
|
||||
23903.5
|
||||
39288.586
|
||||
24230.842
|
||||
33095.27
|
||||
40008.74
|
||||
23399.393
|
||||
20106.33
|
||||
60893.176
|
||||
38653.543
|
||||
18862.43
|
||||
25474.744
|
||||
40467.02
|
||||
31432.373
|
||||
42575.1
|
||||
40211.69
|
||||
23536.877
|
||||
31785.902
|
||||
27242.39
|
||||
58078.03
|
||||
34051.11
|
||||
32087.057
|
||||
54346.332
|
||||
25193.23
|
||||
35360.48
|
||||
53757.113
|
||||
34954.574
|
||||
31432.373
|
||||
25671.148
|
||||
22266.79
|
||||
40597.957
|
||||
40925.297
|
||||
37887.562
|
||||
29926.6
|
||||
40925.297
|
||||
27635.203
|
||||
28813.635
|
||||
31962.668
|
||||
36015.164
|
||||
62896.51
|
||||
55570.594
|
||||
43458.93
|
||||
41579.98
|
||||
20957.42
|
||||
35576.527
|
||||
27556.64
|
||||
40270.613
|
||||
31825.184
|
||||
24885.527
|
||||
36139.55
|
||||
22901.834
|
||||
40205.145
|
||||
31432.373
|
||||
30777.69
|
||||
90353.984
|
||||
67767.37
|
||||
41056.234
|
||||
24950.994
|
||||
23379.754
|
||||
34705.793
|
||||
32741.742
|
||||
34051.11
|
||||
38633.902
|
||||
27700.67
|
||||
30188.47
|
||||
41252.64
|
||||
58850.56
|
||||
58503.58
|
||||
28158.951
|
||||
72022.82
|
||||
36047.9
|
||||
19746.254
|
||||
26194.896
|
||||
59911.15
|
||||
19713.52
|
||||
28217.871
|
||||
34902.2
|
||||
29468.318
|
||||
26784.113
|
||||
30253.941
|
||||
28447.01
|
||||
29998.615
|
||||
32427.494
|
||||
46058.023
|
||||
19517.113
|
||||
25933.021
|
||||
32787.566
|
||||
38542.246
|
||||
39288.586
|
||||
38764.84
|
||||
37010.285
|
||||
24885.527
|
||||
25540.213
|
||||
41449.043
|
||||
21808.51
|
||||
30123.004
|
||||
32741.742
|
||||
34051.11
|
||||
59583.81
|
||||
35353.934
|
||||
22345.35
|
||||
41776.387
|
||||
42169.2
|
||||
56965.07
|
||||
28813.635
|
||||
29075.51
|
||||
72022.82
|
||||
41252.64
|
||||
32931.598
|
||||
20184.893
|
||||
28158.951
|
||||
22960.756
|
||||
43871.38
|
||||
24230.842
|
||||
30777.69
|
||||
91002.125
|
||||
82982.234
|
||||
32440.588
|
||||
66588.94
|
||||
43871.38
|
||||
26194.896
|
||||
31432.373
|
||||
37160.86
|
||||
44080.875
|
||||
13120.846
|
||||
38876.133
|
||||
36669.848
|
||||
34705.793
|
||||
39282.04
|
||||
18796.96
|
||||
46686.523
|
||||
33049.44
|
||||
25540.213
|
||||
24394.514
|
||||
28852.916
|
||||
30777.69
|
||||
30253.941
|
||||
44526.062
|
||||
25088.48
|
||||
24885.527
|
||||
42562.008
|
||||
48454.168
|
||||
34836.73
|
||||
25239.057
|
||||
26325.834
|
||||
46837.098
|
||||
91663.36
|
||||
34705.793
|
||||
43478.566
|
||||
37324.53
|
||||
34005.28
|
||||
31432.373
|
||||
75296.24
|
||||
20905.045
|
||||
38699.367
|
||||
28158.951
|
||||
32741.742
|
||||
28879.102
|
||||
34051.11
|
||||
79355.28
|
||||
38633.902
|
||||
26771.02
|
||||
30974.094
|
||||
49108.855
|
||||
35340.84
|
||||
16806.719
|
||||
27392.97
|
||||
26613.895
|
||||
178474.53
|
||||
32041.23
|
||||
32087.057
|
||||
27438.799
|
||||
44133.254
|
||||
26194.896
|
||||
62202.547
|
||||
48454.168
|
||||
16806.719
|
||||
21350.23
|
||||
94282.09
|
||||
32087.057
|
||||
32198.354
|
||||
27504.266
|
||||
19746.254
|
||||
34705.793
|
||||
26260.363
|
||||
36872.8
|
||||
74641.555
|
||||
28813.635
|
||||
28813.635
|
||||
32087.057
|
||||
46116.945
|
||||
24558.186
|
||||
28158.951
|
||||
46837.098
|
||||
35916.96
|
||||
51275.86
|
||||
141419.39
|
||||
36669.848
|
||||
34116.58
|
||||
39943.27
|
||||
43871.38
|
||||
32087.057
|
||||
29514.146
|
||||
41907.324
|
||||
39484.992
|
||||
20302.736
|
||||
24885.527
|
||||
28263.7
|
||||
54542.74
|
||||
75754.516
|
||||
20302.736
|
||||
33468.44
|
||||
26980.518
|
||||
48454.168
|
||||
52513.21
|
||||
41514.516
|
||||
49684.977
|
||||
35916.96
|
||||
16440.096
|
||||
26404.395
|
||||
26856.127
|
||||
50765.207
|
||||
40205.145
|
||||
38175.62
|
||||
28715.432
|
||||
30646.752
|
||||
43871.38
|
||||
49763.54
|
||||
44526.062
|
||||
43871.38
|
||||
28695.791
|
||||
57619.754
|
||||
49783.18
|
||||
54018.99
|
||||
37979.215
|
||||
23445.22
|
||||
35818.758
|
||||
41907.324
|
||||
31432.373
|
||||
32741.742
|
||||
49108.855
|
||||
43216.69
|
||||
32853.035
|
||||
16806.719
|
||||
30777.69
|
||||
30777.69
|
||||
42496.543
|
||||
31432.373
|
||||
43871.38
|
||||
24257.031
|
||||
21494.262
|
||||
35353.934
|
||||
47642.363
|
||||
63053.637
|
||||
31694.248
|
||||
51727.594
|
||||
24885.527
|
||||
31432.373
|
||||
26522.238
|
||||
41514.516
|
||||
23183.348
|
||||
26129.43
|
||||
27609.016
|
||||
27045.986
|
||||
32087.057
|
||||
59583.81
|
||||
22201.322
|
||||
37913.75
|
||||
32087.057
|
||||
28394.637
|
||||
35360.48
|
||||
42562.008
|
||||
25186.682
|
||||
30646.752
|
||||
35583.07
|
||||
58929.12
|
||||
55917.574
|
||||
32663.182
|
||||
31432.373
|
||||
20152.158
|
||||
30273.582
|
||||
32087.057
|
||||
33396.426
|
||||
26987.064
|
||||
22986.941
|
||||
34705.793
|
||||
43871.38
|
||||
40139.676
|
||||
59321.934
|
||||
42889.35
|
||||
24230.842
|
||||
56572.258
|
||||
44480.234
|
||||
48126.83
|
||||
22921.475
|
||||
27635.203
|
||||
34116.58
|
||||
43544.035
|
||||
20152.158
|
||||
39347.508
|
||||
28741.62
|
||||
37782.812
|
||||
34483.2
|
||||
66588.94
|
||||
51596.656
|
||||
30646.752
|
||||
46837.098
|
||||
35615.805
|
||||
68507.164
|
||||
27504.266
|
||||
40467.02
|
||||
22921.475
|
||||
52382.277
|
||||
13101.205
|
||||
31157.406
|
||||
67073.41
|
||||
26194.896
|
||||
28813.635
|
||||
98210.2
|
||||
30692.578
|
||||
23144.066
|
||||
33913.625
|
||||
38005.406
|
||||
43871.38
|
||||
31753.17
|
||||
44787.938
|
||||
77914.98
|
||||
31432.373
|
||||
34902.2
|
||||
31432.373
|
||||
93987.484
|
||||
23314.283
|
||||
17094.781
|
||||
29461.773
|
||||
51727.594
|
||||
32741.742
|
||||
27222.752
|
||||
34083.844
|
||||
26319.285
|
||||
74137.45
|
||||
23968.969
|
||||
80991.99
|
||||
37802.453
|
||||
32080.512
|
||||
57861.99
|
||||
31596.045
|
||||
28106.574
|
||||
44067.785
|
||||
44067.785
|
||||
66588.94
|
||||
45599.746
|
||||
29271.914
|
||||
32663.182
|
||||
35916.96
|
||||
30391.424
|
||||
32087.057
|
||||
22162.04
|
||||
31098.484
|
||||
39288.586
|
||||
32388.213
|
||||
49914.117
|
||||
34993.855
|
||||
34993.855
|
||||
51727.594
|
||||
26522.238
|
||||
56978.16
|
||||
93169.13
|
||||
44774.84
|
||||
38142.887
|
||||
32453.682
|
||||
19667.691
|
||||
23563.064
|
||||
22901.834
|
||||
32126.34
|
||||
50228.363
|
||||
47557.25
|
||||
43111.94
|
||||
|
|
Loading…
Reference in New Issue
Block a user