def test_WoeManager(self):
aa = AreaAnalyst(self.sites, self.sites)
w1 = WoeManager([self.factor], aa)
p = w1.getPrediction(self.sites).getBand(1)
assert_array_equal(p, self.sites.getBand(1))
initState = Raster("../../examples/data.tif")
finalState = Raster("../../examples/data1.tif")
aa = AreaAnalyst(initState, finalState)
w = WoeManager([initState], aa)
p = w.getPrediction(initState).getBand(1)
# Calculate by hands:
# 1->1 transition raster:
r11 = [[1, 1, 0, 0], [0, 1, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]
# 1->2 raster:
r12 = [[0, 0, 1, 0], [0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]
# 1->3 raster:
r13 = [[0, 0, 0, 1], [0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]
# 2->1
r21 = [[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]
# 2->2
r22 = [[0, 0, 0, 0], [0, 0, 1, 0], [0, 0, 0, 0], [0, 0, 0, 0]]
# 2->3
r23 = [[0, 0, 0, 0], [0, 0, 0, 1], [1, 1, 1, 1], [0, 0, 0, 0]]
# 3->1
r31 = [[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0], [1, 1, 0, 0]]
# 3->2
r32 = [[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 1]]
# 3->3
r33 = [[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 1, 0]]
geodata = initState.getGeodata()
sites = {"11": r11, "12": r12, "13": r13, "21": r21, "22": r22, "23": r23, "31": r31, "32": r32, "33": r33}
woeDict = {} # WoE of transitions
for k in sites.keys(): #
if k != "21": # !!! r21 is zero
x = Raster()
x.create([np.ma.array(data=sites[k])], geodata)
sites[k] = x
woeDict[k] = woe(initState.getBand(1), x.getBand(1))
# w1max = np.maximum(woeDict['11'], woeDict['12'], woeDict['13'])
# w2max = np.maximum(woeDict['22'], woeDict['23'])
# w3max = np.maximum(woeDict['31'], woeDict['32'], woeDict['33'])
# Answer is index of finalClass that maximizes weights of transiotion initClass -> finalClass
answer = [[1, 1, 1, 1], [1, 1, 3, 3], [3, 3, 3, 3], [1, 1, 1, 1]]
assert_array_equal(p, answer)
w = WoeManager([initState], aa, bins={0: [[2]]})
p = w.getPrediction(initState).getBand(1)
def test_WoeManager(self):
aa = AreaAnalyst(self.sites, self.sites)
w1 = WoeManager([self.factor], aa)
w1.train()
p = w1.getPrediction(self.sites).getBand(1)
answer = [[0,3,0], [0,3,0], [9,0,3]]
answer = ma.array(data = answer, mask = self.mask)
assert_array_equal(p, answer)
initState = Raster('../../examples/data.tif')
#~ [1,1,1,1],
#~ [1,1,2,2],
#~ [2,2,2,2],
#~ [3,3,3,3]
finalState = Raster('../../examples/data1.tif')
#~ [1,1,2,3],
#~ [3,1,2,3],
#~ [3,3,3,3],
#~ [1,1,3,2]
aa = AreaAnalyst(initState, finalState)
w = WoeManager([initState], aa)
w.train()
#print w.woe
p = w.getPrediction(initState).getBand(1)
self.assertEquals(p.dtype, np.uint8)
# Calculate by hands:
#1->1 transition raster:
r11 = [
[1, 1, 0, 0],
[0, 1, 0, 0],
[0, 0, 0, 0],
[0, 0, 0, 0]
]
#1->2 raster:
r12 = [
[0, 0, 1, 0],
[0, 0, 0, 0],
[0, 0, 0, 0],
[0, 0, 0, 0]
]
#1->3 raster:
r13 = [
[0, 0, 0, 1],
[0, 0, 0, 0],
[0, 0, 0, 0],
[0, 0, 0, 0]
]
# 2->1
r21 = [
[0, 0, 0, 0],
[0, 0, 0, 0],
[0, 0, 0, 0],
[0, 0, 0, 0]
]
# 2->2
r22 = [
[0, 0, 0, 0],
[0, 0, 1, 0],
[0, 0, 0, 0],
[0, 0, 0, 0]
]
# 2->3
r23 = [
[0, 0, 0, 0],
[0, 0, 0, 1],
[1, 1, 1, 1],
[0, 0, 0, 0]
]
# 3->1
r31 = [
[0, 0, 0, 0],
[0, 0, 0, 0],
[0, 0, 0, 0],
[1, 1, 0, 0]
]
# 3->2
r32 = [
[0, 0, 0, 0],
[0, 0, 0, 0],
[0, 0, 0, 0],
[0, 0, 0, 1]
]
# 3->3
r33 = [
[0, 0, 0, 0],
[0, 0, 0, 0],
[0, 0, 0, 0],
[0, 0, 1, 0]
]
geodata = initState.getGeodata()
sites = {'11': r11, '12': r12, '13': r13, '21': r21, '22': r22, '23': r23, '31': r31, '32': r32, '33': r33}
woeDict = {} # WoE of transitions
for k in sites.keys(): #
if k !='21' : # !!! r21 is zero
x = Raster()
x.create([np.ma.array(data=sites[k])], geodata)
sites[k] = x
woeDict[k] = woe(initState.getBand(1), x.getBand(1))
#w1max = np.maximum(woeDict['11'], woeDict['12'], woeDict['13'])
#w2max = np.maximum(woeDict['22'], woeDict['23'])
#w3max = np.maximum(woeDict['31'], woeDict['32'], woeDict['33'])
# Answer is a transition code with max weight
answer = [
[0, 0, 0, 0],
[0, 0, 5, 5],
[5, 5, 5, 5],
[6, 6, 6, 6]
]
assert_array_equal(p, answer)
w = WoeManager([initState], aa, bins = {0: [[2], ],})
w.train()
p = w.getPrediction(initState).getBand(1)
self.assertEquals(p.dtype, np.uint8)
c = w.getConfidence().getBand(1)
self.assertEquals(c.dtype, np.uint8)
def test_CheckBins(self):
aa = AreaAnalyst(self.sites, self.sites)
w1 = WoeManager([self.factor], aa, bins = None)
self.assertTrue(w1.checkBins())
w1 = WoeManager([self.factor], aa, bins = {0: [None]})
self.assertTrue(w1.checkBins())
w1 = WoeManager([self.factor], aa, bins = {0: [[1, 2, 3]]})
self.assertTrue(w1.checkBins())
w1 = WoeManager([self.factor], aa, bins = {0: [[1, 4]]})
self.assertFalse(w1.checkBins())
w1 = WoeManager([self.factor], aa, bins = {0: [[-1, 1]]})
self.assertFalse(w1.checkBins())
w1 = WoeManager([self.factor], aa, bins = {0: [[2, 3, 1]]})
self.assertFalse(w1.checkBins())
def test_WoeManager(self):
aa = AreaAnalyst(self.sites, self.sites)
w1 = WoeManager([self.factor], aa)
w1.train()
p = w1.getPrediction(self.sites).getBand(1)
answer = [[0, 3, 0], [0, 3, 0], [9, 0, 3]]
answer = ma.array(data=answer, mask=self.mask)
assert_array_equal(p, answer)
initState = Raster('../../examples/data.tif')
#~ [1,1,1,1],
#~ [1,1,2,2],
#~ [2,2,2,2],
#~ [3,3,3,3]
finalState = Raster('../../examples/data1.tif')
#~ [1,1,2,3],
#~ [3,1,2,3],
#~ [3,3,3,3],
#~ [1,1,3,2]
aa = AreaAnalyst(initState, finalState)
w = WoeManager([initState], aa)
w.train()
#print w.woe
p = w.getPrediction(initState).getBand(1)
self.assertEquals(p.dtype, np.uint8)
# Calculate by hands:
#1->1 transition raster:
r11 = [[1, 1, 0, 0], [0, 1, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]
#1->2 raster:
r12 = [[0, 0, 1, 0], [0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]
#1->3 raster:
r13 = [[0, 0, 0, 1], [0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]
# 2->1
r21 = [[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]
# 2->2
r22 = [[0, 0, 0, 0], [0, 0, 1, 0], [0, 0, 0, 0], [0, 0, 0, 0]]
# 2->3
r23 = [[0, 0, 0, 0], [0, 0, 0, 1], [1, 1, 1, 1], [0, 0, 0, 0]]
# 3->1
r31 = [[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0], [1, 1, 0, 0]]
# 3->2
r32 = [[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 1]]
# 3->3
r33 = [[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 1, 0]]
geodata = initState.getGeodata()
sites = {
'11': r11,
'12': r12,
'13': r13,
'21': r21,
'22': r22,
'23': r23,
'31': r31,
'32': r32,
'33': r33
}
woeDict = {} # WoE of transitions
for k in sites.keys(): #
if k != '21': # !!! r21 is zero
x = Raster()
x.create([np.ma.array(data=sites[k])], geodata)
sites[k] = x
woeDict[k] = woe(initState.getBand(1), x.getBand(1))
#w1max = np.maximum(woeDict['11'], woeDict['12'], woeDict['13'])
#w2max = np.maximum(woeDict['22'], woeDict['23'])
#w3max = np.maximum(woeDict['31'], woeDict['32'], woeDict['33'])
# Answer is a transition code with max weight
answer = [[0, 0, 0, 0], [0, 0, 5, 5], [5, 5, 5, 5], [6, 6, 6, 6]]
assert_array_equal(p, answer)
w = WoeManager([initState], aa, bins={
0: [
[2],
],
})
w.train()
p = w.getPrediction(initState).getBand(1)
self.assertEquals(p.dtype, np.uint8)
c = w.getConfidence().getBand(1)
self.assertEquals(c.dtype, np.uint8)
def test_CheckBins(self):
aa = AreaAnalyst(self.sites, self.sites)
w1 = WoeManager([self.factor], aa, bins=None)
self.assertTrue(w1.checkBins())
w1 = WoeManager([self.factor], aa, bins={0: [None]})
self.assertTrue(w1.checkBins())
w1 = WoeManager([self.factor], aa, bins={0: [[1, 2, 3]]})
self.assertTrue(w1.checkBins())
w1 = WoeManager([self.factor], aa, bins={0: [[1, 4]]})
self.assertFalse(w1.checkBins())
w1 = WoeManager([self.factor], aa, bins={0: [[-1, 1]]})
self.assertFalse(w1.checkBins())
w1 = WoeManager([self.factor], aa, bins={0: [[2, 3, 1]]})
self.assertFalse(w1.checkBins())
