diff --git a/config.txt b/config.txt index 6ad06eb..3c840d1 100644 --- a/config.txt +++ b/config.txt @@ -1 +1 @@ ---precision 1 +--metric RMSE --precision 1 diff --git a/dev-0/out.tsv b/dev-0/out.tsv index 5201481..1f46d6a 100644 --- a/dev-0/out.tsv +++ b/dev-0/out.tsv @@ -1,4 +1,3 @@ -0 77497.59817308374 12145.183204738423 49614.760815142654 diff --git a/main.py b/main.py index 1ea9897..9fae238 100644 --- 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) diff --git a/test-A/out.tsv b/test-A/out.tsv index e3a58cd..aee9d0b 100644 --- a/test-A/out.tsv +++ b/test-A/out.tsv @@ -1,4 +1,3 @@ -0 24338.076058211736 23636.277893734165 12031.8488286715