AutoML conversion

convert.py

# write training data function
def write_training_data_to_zipfile(path, Xdata, Ydata):
  os.chdir(path)
  count = {}
  num_rows = Xdata.shape[0]
  for i in range(1,num_rows):
    label = Ydata.iloc[i]
    feature = Xdata.iloc[i]
    folder = path + "/" + label
    if not os.path.exists(folder):
      os.mkdir(folder)
    if not (label in count.keys()):
      count[label] = 1
    else:
      count[label] += 1
    file = folder + "/" + "sample" + str(count[label]) + ".txt"
    with open(file, 'wb') as fp:
      fp.write(feature.encode("utf-8"))
      fp.write("\n".encode('utf-8'))
      fp.close()
      
! zip -r documents.zip *