s437622/auta-public
1
0
Fork 0
forked from kubapok/auta-public
Code Issues Pull Requests Projects Releases Wiki Activity

change config.txt

Browse Source
This commit is contained in:
JPogodzinski 2021-05-01 12:57:20 +02:00
parent 11ce81b395
commit a0e14fe99d
4 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config.txt
View File

@ -1 +1 @@
--precision 1 --metric RMSE --precision 1

1
dev-0/out.tsv
View File

@ -1,4 +1,3 @@
0
77497.59817308374 77497.59817308374
12145.183204738423 12145.183204738423
49614.760815142654 49614.760815142654

1 0 77497.59817308374
0
1 77497.59817308374 77497.59817308374
2 12145.183204738423 12145.183204738423
3 49614.760815142654 49614.760815142654

5
main.py
View File

@ -35,7 +35,8 @@ cars_dev.drop('brand', inplace=True, axis=1)
X_dev = pd.DataFrame(cars_dev) X_dev = pd.DataFrame(cars_dev)
Y_dev_predicted = model.predict(X_dev) Y_dev_predicted = model.predict(X_dev)
pd.DataFrame(Y_dev_predicted).to_csv('dev-0/out.tsv', sep='\t', index=False) print(Y_dev_predicted)
pd.DataFrame(Y_dev_predicted).to_csv('dev-0/out.tsv', sep='\t', index=False, header=False)
cars_test=pd.read_csv('test-A/in.tsv', sep='\t', names=names) cars_test=pd.read_csv('test-A/in.tsv', sep='\t', names=names)
@ -44,7 +45,7 @@ cars_test.drop('brand', inplace=True, axis=1)
X_test = pd.DataFrame(cars_test) X_test = pd.DataFrame(cars_test)
Y_test_predicted = model.predict(X_test) Y_test_predicted = model.predict(X_test)
pd.DataFrame(Y_test_predicted).to_csv('test-A/out.tsv', sep='\t', index=False) pd.DataFrame(Y_test_predicted).to_csv('test-A/out.tsv', sep='\t', index=False, header=False)
error = np.sqrt(mean_squared_error(Y_dev, Y_dev_predicted)) error = np.sqrt(mean_squared_error(Y_dev, Y_dev_predicted))
print(error) print(error)

1
test-A/out.tsv
View File

@ -1,4 +1,3 @@
0
24338.076058211736 24338.076058211736
23636.277893734165 23636.277893734165
12031.8488286715 12031.8488286715

1 0 24338.076058211736
0
1 24338.076058211736 24338.076058211736
2 23636.277893734165 23636.277893734165
3 12031.8488286715 12031.8488286715