change config.txt
This commit is contained in:
parent
11ce81b395
commit
a0e14fe99d
@ -1 +1 @@
|
|||||||
--precision 1
|
--metric RMSE --precision 1
|
||||||
|
@ -1,4 +1,3 @@
|
|||||||
0
|
|
||||||
77497.59817308374
|
77497.59817308374
|
||||||
12145.183204738423
|
12145.183204738423
|
||||||
49614.760815142654
|
49614.760815142654
|
||||||
|
|
5
main.py
5
main.py
@ -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,4 +1,3 @@
|
|||||||
0
|
|
||||||
24338.076058211736
|
24338.076058211736
|
||||||
23636.277893734165
|
23636.277893734165
|
||||||
12031.8488286715
|
12031.8488286715
|
||||||
|
|
Loading…
Reference in New Issue
Block a user