16 lines
436 B
Python
16 lines
436 B
Python
|
from sklearn.metrics import mean_squared_error
|
||
|
import pandas as pd
|
||
|
|
||
|
def RMSE(exp, pred):
|
||
|
expected = exp
|
||
|
predicted = pred
|
||
|
|
||
|
data_exp = pd.read_csv(expected, header=0, sep='\t')
|
||
|
data_pred = pd.read_csv(predicted, header=0, sep='\t')
|
||
|
|
||
|
rmse = mean_squared_error(data_exp, data_pred, squared=False)
|
||
|
|
||
|
return rmse
|
||
|
|
||
|
print(RMSE("dev-0/expected.tsv", "dev-0/out.tsv"))
|
||
|
print(RMSE("dev-1/expected.tsv", "dev-1/out.tsv"))
|