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