change config.txt

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

View File

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

View File

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

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

View File

@ -35,7 +35,8 @@ cars_dev.drop('brand', inplace=True, axis=1)
X_dev = pd.DataFrame(cars_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)
@ -44,7 +45,7 @@ cars_test.drop('brand', inplace=True, axis=1)
X_test = pd.DataFrame(cars_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))
print(error)

View File

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

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