Skip to content
GitLab
Explore
Sign in
Primary navigation
Search or go to…
Project
iamap
Manage
Activity
Members
Labels
Plan
Issues
Issue boards
Milestones
Wiki
Code
Merge requests
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Snippets
Build
Pipelines
Jobs
Pipeline schedules
Artifacts
Deploy
Releases
Package Registry
Container Registry
Model registry
Operate
Environments
Terraform modules
Monitor
Incidents
Analyze
Value stream analytics
Contributor analytics
CI/CD analytics
Repository analytics
Model experiments
Help
Help
Support
GitLab documentation
Compare GitLab plans
Community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
AMAP
iamap
Commits
9bc92d28
Commit
9bc92d28
authored
5 months ago
by
paul.tresson_ird.fr
Browse files
Options
Downloads
Patches
Plain Diff
handle input vector file more properly, clean imports
parent
b4d26590
No related branches found
No related tags found
No related merge requests found
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
ml.py
+5
-22
5 additions, 22 deletions
ml.py
utils/algo.py
+3
-2
3 additions, 2 deletions
utils/algo.py
with
8 additions
and
24 deletions
ml.py
+
5
−
22
View file @
9bc92d28
import
os
import
ast
import
numpy
as
np
from
pathlib
import
Path
from
typing
import
Dict
,
Any
import
joblib
import
json
import
tempfile
import
rasterio
from
rasterio
import
windows
...
...
@@ -13,30 +11,15 @@ import geopandas as gpd
import
pandas
as
pd
from
shapely.geometry
import
box
from
qgis.PyQt.QtCore
import
QCoreApplication
from
qgis.core
import
(
Qgis
,
QgsGeometry
,
from
qgis.core
import
(
QgsProcessingParameterBoolean
,
QgsProcessingParameterFile
,
QgsProcessingParameterEnum
,
QgsCoordinateTransform
,
QgsProcessingException
,
QgsProcessingAlgorithm
,
QgsProcessingParameterRasterLayer
,
QgsProcessingParameterFolderDestination
,
QgsProcessingParameterVectorLayer
,
QgsProcessingParameterString
,
QgsProcessingParameterBand
,
QgsProcessingParameterNumber
,
QgsProcessingParameterExtent
,
QgsProcessingParameterCrs
,
QgsProcessingParameterDefinition
,
QgsProcessingParameterDefinition
)
import
torch
import
torch.nn
as
nn
from
sklearn.ensemble
import
RandomForestClassifier
from
sklearn.preprocessing
import
LabelEncoder
from
sklearn.model_selection
import
train_test_split
from
.utils.misc
import
get_unique_filename
from
.utils.geo
import
get_random_samples_in_gdf
,
get_unique_col_name
from
.utils.algo
import
(
SHPAlgorithm
,
...
...
@@ -120,7 +103,7 @@ class MLAlgorithm(SHPAlgorithm):
self
.
addParameter
(
QgsProcessingParameter
File
(
QgsProcessingParameter
VectorLayer
(
name
=
self
.
TEMPLATE
,
description
=
self
.
tr
(
'
Input shapefile path for training data set for random forest (if no test data_set, will be devised in train and test)
'
),
...
...
@@ -129,7 +112,7 @@ class MLAlgorithm(SHPAlgorithm):
)
self
.
addParameter
(
QgsProcessingParameter
File
(
QgsProcessingParameter
VectorLayer
(
name
=
self
.
TEMPLATE_TEST
,
description
=
self
.
tr
(
'
Input shapefile path for test dataset.
'
),
...
...
This diff is collapsed.
Click to expand it.
utils/algo.py
+
3
−
2
View file @
9bc92d28
...
...
@@ -18,10 +18,11 @@ from qgis.core import (Qgis,
QgsProcessingParameterNumber
,
QgsProcessingParameterEnum
,
QgsProcessingParameterFile
,
QgsProcessingParameterVectorLayer
,
QgsProcessingParameterExtent
,
QgsProcessingParameterString
,
QgsProcessingParameterCrs
,
QgsProcessingParameterDefinition
,
QgsProcessingParameterDefinition
,
QgsProcessingParameterVectorLayer
,
)
import
rasterio
from
rasterio
import
windows
...
...
@@ -903,7 +904,7 @@ class SHPAlgorithm(IAMAPAlgorithm):
)
self
.
addParameter
(
QgsProcessingParameter
File
(
QgsProcessingParameter
VectorLayer
(
name
=
self
.
TEMPLATE
,
description
=
self
.
tr
(
'
Input shapefile path
'
),
...
...
This diff is collapsed.
Click to expand it.
Preview
0%
Loading
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Save comment
Cancel
Please
register
or
sign in
to comment