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")) |