def get_stockholm_run1(self):
self.store_path = self.store_pathv.get()
if (self.v_num.get() == 1):
self.num_stock = 'all'
else:
self.num_stock = self.num.get()
if (self.v.get() == 1):
self.output_type = 'json'
elif (self.v.get() == 2):
self.output_type = 'csv'
else:
self.output_type = 'all'
if (self.v2.get() == 1):
self.charset = 'utf-8'
else:
self.charset = 'gbk'
self.start_date = self.start_date1.get() + self.start_date2.get(
) + self.start_date3.get()
self.end_date = self.end_date1.get() + self.end_date2.get(
) + self.end_date3.get()
self.export_file_name = self.export_en.get()
self.stock_en.set('正在获取数据')
stockh = Stockholm(self)
stockh.run1()
self.stock_en.set('获取数据完成')
def get_learning_test(self):
if (self.store_path == 'USER_HOME/tmp/stockholm_export'):
self.export_folder = os.path.expanduser(
'~') + '/tmp/stockholm_export'
else:
self.export_folder = self.store_path
self.learning_date = self.learning_date1.get(
) + self.learning_date2.get() + self.learning_date3.get()
self.num_stock = self.learning_symbol.get()
#print(self.num_stock)
self.export_file_name = 'learning1'
self.start_date = '20110101'
self.end_date = self.learning_date
stockh = Stockholm(self)
stockh.run1()
self.export_file_name = 'learning2'
ld = datetime.datetime(int(self.learning_date1.get()),
int(self.learning_date2.get()),
int(self.learning_date3.get()))
self.start_date = (ld + datetime.timedelta(days=1)).strftime("%Y%m%d")
self.end_date = get_date_str(None)
stockh2 = Stockholm(self)
stockh2.run1()
self.variance = learningtest.stock_predection_test(
self.export_folder + '/' + "learning1.csv",
self.export_folder + '/' + "learning2.csv", 60)
def get_learning(self):
if (self.store_path == 'USER_HOME/tmp/stockholm_export'):
self.export_folder = os.path.expanduser(
'~') + '/tmp/stockholm_export'
else:
self.export_folder = self.store_path
self.learning_date = self.learning_date1.get(
) + self.learning_date2.get() + self.learning_date3.get()
self.num_stock = self.learning_symbol.get()
#print(self.num_stock)
self.export_file_name = 'learning'
self.start_date = '20120101'
self.end_date = get_date_str(None)
stockh3 = Stockholm(self)
stockh3.run1()
self.predicted_price = learning.stock_predection(self.export_folder +
'/' + "learning.csv")