change config.txt

2021-05-01 12:57:20 +02:00 · 2021-05-01 12:57:20 +02:00 · a0e14fe99d
commit a0e14fe99d
parent 11ce81b395
4 changed files with 4 additions and 5 deletions
--- a/config.txt
+++ b/config.txt
@ -1 +1 @@
--precision 1
+--metric RMSE --precision 1
--- a/dev-0/out.tsv
+++ b/dev-0/out.tsv
@ -1,4 +1,3 @@
-0
 77497.59817308374
 12145.183204738423
 49614.760815142654
--- a/main.py
+++ b/main.py
@ -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)
--- a/test-A/out.tsv
+++ b/test-A/out.tsv
@ -1,4 +1,3 @@
-0
 24338.076058211736
 23636.277893734165
 12031.8488286715