yufeng0528 4 years ago
parent
commit
35341653a2
2 changed files with 5 additions and 0 deletions
  1. 3 0
      bbztx/get_data.py
  2. 2 0
      linear/train.py

+ 3 - 0
bbztx/get_data.py

@@ -1,5 +1,8 @@
1 1
 #!/usr/bin/python
2 2
 # -*- coding:utf-8 -*-
3
+import sys
4
+import os
5
+sys.path.append(os.path.abspath('..'))
3 6
 from util import pgsql_util
4 7
 import json
5 8
 

+ 2 - 0
linear/train.py

@@ -4,6 +4,8 @@
4 4
 最简单的mse
5 5
 '''
6 6
 import sys
7
+import os
8
+sys.path.append(os.path.abspath('..'))
7 9
 
8 10
 import numpy as np
9 11
 from sklearn.linear_model import LinearRegression