This commit is contained in:
Mikołaj Pokrywka 2022-06-05 17:01:04 +02:00
parent 4dd3799dbb
commit 8216b1705f
2 changed files with 1 additions and 5 deletions

View File

@ -1,4 +0,0 @@
outs:
- md5: b57fd92c7032152f66ce5d169c2c3fdc
size: 6994695
path: data_test.csv

View File

@ -17,7 +17,7 @@ if __name__ == "__main__":
company_profile = vectorizer.fit_transform(company_profile) company_profile = vectorizer.fit_transform(company_profile)
model = torch.load('model') model = torch.load('model')
data_test = pd.read_csv('data_test.csv', engine='python', header=None) data_test = data[1:100]
data_test = data_test.dropna() data_test = data_test.dropna()
x_test = data_test[5] x_test = data_test[5]
y_test = data_test[17] y_test = data_test[17]