Skip to content

Training with AI

Safin Singh edited this page Apr 26, 2020 · 1 revision
  • Run the init function and pass in the path with your data. In this data folder, there should be multiple folders representing the desired categories

Example: init('../data/train/')

  • Run the train1 function passing in what you would like to save this as. This will print data which is useful for train2

Example: train1('train1')

  • Run the train2 function passing in what you would like to save this as. Also, pass in the slice that you would like to train your machine on (min and then max) (e.x. 1e-6 and 1e-5) and finally pass in where you would like to export your pickle file

Example: train2('train2',1e-6,1e-5,'../exports/')

Clone this wiki locally