import file_handler
import random
import FCM
import numpy as np
if __name__ == '__main__':
data = file_handler.read_file('input_data/2clstrain1200.csv')
input_size = len(data)
train_data = []
test_data = []
# producing random data for final drawing X:[-5, 17.5], Y:[-2.5, 15]
X_rand, Y_rand = FCM.produce_random_num(-8, 19, -4, 17, 2000)
random_input = np.concatenate((X_rand, Y_rand), axis=0)
# Here we should choose randomly train and test data
train_size = int(70 * input_size / 100)
test_size = input_size - train_size
train_data = [data[i] for i in range(test_size, input_size)]
test_data = [data[i] for i in range(0, test_size)]
train_data_no_label = []
train_labels = []
test_data_no_label = []
test_labels = []
for i in range(len(train_data)):
train_data_no_label.append([train_data[i][0], train_data[i][1]])
train_labels.append(train_data[i][2])
