CONDA small refactoring

This commit is contained in:
Mikołaj Pokrywka 2022-05-23 00:10:14 +02:00
parent cee02a2d2b
commit 2b8884af80

View File

@ -46,11 +46,11 @@ def train(epochs):
data = data.dropna()
company_profile = data["company_profile"]
# data_train, data_test = train_test_split(data, test_size=3000, random_state=1)
# data_dev, data_test = train_test_split(data_test, test_size=1500, random_state=1)
data_train = pd.read_csv('data_train.csv', engine='python', header=None).dropna()
data_dev = pd.read_csv('data_dev.csv', engine='python', header=None).dropna()
data_test = pd.read_csv('data_test.csv', engine='python', header=None).dropna()
data_train, data_test = train_test_split(data, test_size=3000, random_state=1)
data_dev, data_test = train_test_split(data_test, test_size=1500, random_state=1)
# data_train = pd.read_csv('data_train.csv', engine='python', header=None).dropna()
# data_dev = pd.read_csv('data_dev.csv', engine='python', header=None).dropna()
# data_test = pd.read_csv('data_test.csv', engine='python', header=None).dropna()
x_train = data_train[5]
x_dev = data_dev[5]